作者 | 梁唐       责编 | 欧阳姝黎

今天给大家聊聊C++中的头文件,之前我在写算法专题展示源代码的时候,很多小伙伴给我留言说被我的头文件中的内容震惊了。其实之所以我的头文件这么复杂,完全是因为它是我一直从大学时期acm竞赛当中沿用下来的。对于acm竞赛的选手们来说,这样的头文件其实算是小儿科了。

今天就和大家来看看,acmer 的头文件当中都藏着哪些秘密。

首先,我们先来看我完整的头文件代码。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cstdlib>
#include <cmath>
#include <queue>
#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <functional>
#define rep(i,a,b) for (int i=a;i<b;i++)
#define Rep(i,a,b) for (int i=a;i>b;i--)
#define foreach(e,x) for (__typeof(x.begin()) e=x.begin();e!=x.end();e++)
#define mid ((l+r)>>1)
#define lson (k<<1)
#define rson (k<<1|1)
#define MEM(a,x) memset(a,x,sizeof a)
#define pii pair<int, int>
#define LL long long
using namespace std;
const int N=500005;
const long long Mod=99997867;

include部分

首先我们来看 include 的部分,我们一个一个来看,iostream 不用多说了,C++标准输入输出的头文件,包含了 C++输入输出流函数,也就是经典的 cin、cout。说到 cin、cout 多说两句,cin、cout 的开销要比 C 语言下的 scanf 和 printf 慢很多,很容易影响程序运行的性能。所以对于 acmer 来说,能用 scanf 和 printf 完成的,就绝对不会用 cin、cout。当然 scanf 和 printf 也不是最快的,还有更快的 getchar 和 putchar,所以有些玩家会自己手写循环读入 char 然后转成 int 或者是 float 的函数,当然绝大多数情况下并不需要这样。

cstdio 和 iostream 同样的功能,不过是 C 语言中的输入输出函数,不多说。

cstring 同样属于 C 语言,是 C 语言中的字符串库,有很多字符串相关的函数。同样由于性能的原因,能用 C 语言中 char[]完成的就不用使用 C++的 string。接下来的 string 库也不用多说,C++的字符串处理库。

cstdlib 库函数等同于 C 语言中 stdlib.h,封装了一些常用的库函数,如 rand、srand、free、malloc 等。

cmath 库等于 C 语言中的 math.h,封装了一些数学运算相关的库函数,如 pow、sqrt 等。

后面的 queue、vector、map、set 都是 STL 库,包含了一些比较好用的数据结构。比如 queue 中封装了 queue 以及 dequeue,和 priority_queue,也就是队列、双端队列和优先队列。vector、map、set 分别是线性表、映射表以及集合,熟练使用它们可以极大的降低编码的复杂度。

algorithm 库翻译过来就是算法库,当中自然封装了不少的算法。比如 sort 排序、reverse 翻转、next_permutation 下一个全排列、lower_bound、upper_bound 函数等。

最后一个 functional 库,用得不多,顾名思义当中封装了一些关于函数对象的操作。比如 bind、reference_wrapper 等等。

其实关于 include,有一个取巧的方法可以一次性 include 所有的头文件:

#include <bits/stdc++.h>

不过这种方法有一个小问题,并不是所有的环境都支持,尤其是在正式比赛的场合,所以一般情况下大家都是一些网络赛才会用它,我个人由于懒得区分环境,所以还是习惯自己一个一个 include。

define部分

define 是 C++ 当中非常强大的功能,它可以定义规则对代码进行替换。熟练使用 define 同样可以大大简化编码。但是要注意,凡事不能过度,如果 define 使用过多会影响程序的可读性,也可能对其他人的编码造成影响。所以很多大公司是禁止使用 define 的,我个人倒觉得其实也不用这么严肃,define 可以用,遵守规范,适当使用就可以了。

首先是这两行,是 for-loop 的 define。

#define rep(i,a,b) for (int i=a;i<b;i++)
#define Rep(i,a,b) for (int i=a;i>b;i--)
#define foreach(e,x) for (__typeof(x.begin()) e=x.begin();e!=x.end();e++)

rep 是 repeat 的缩写,使用的时候只需要 rep(i, a, b)就可以代替冗长的 for 循环的编写,其中 i 是循环变量,a 和 b 分别是循环的上下界,注意是左闭右开区间。

Rep 也是同样的逻辑,只不过是倒序的循环。

foreach 使用的是 C++11 的新特性,可以实现自动迭代,用的不多,在一些场景下非常方便。

#define mid ((l+r)>>1)

这行用在二分查找当中,左边界是 l,右边界是 r,那么它们的中点就是(l + r) / 2,用位运算表示就是:(l + r) >> 1。例子:

# define之前
while (l + 1 < r) {
    int m = (l + r) >> 1;
    if (a[m] <= v) {
  l = m;
    }else {
  r = m;
    }
}

# define之后
while (l + 1 < r) {
    if (a[mid] <= v) {
  l = mid;
    }else {
  r = mid;
    }
}

#define lson (k<<1)
#define rson (k<<1|1)

这两行主要是用在线段树上,因为 C++ 往往不使用类来实现线段树,而是通过数组来模拟实现。在线段树当中,如果某一个节点的 id 是 u,那么它的左孩子是 2 x u,右孩子的 id 是 2 x u + 1,用位运算来表示就是 u << 1 和 u << 1 | 1。

#define MEM(a,x) memset(a,x,sizeof a)
#define pii pair<int, int>
#define LL long long 

最后三行放在一起说,第一行是 memset 的缩写,memset 可以用来初始化数组。pii 缩写了 pair<int, int>,pair 是用来绑定两个变量的数据结构。最后是 LL 缩写了 long long,long long 是 64 位的 int,它的范围要比 int 大得多。

关于类型的重定义,这里使用 define 并不是非常好,更好的方法是使用 typedef,例如上面的 pair 和 long long 可以写成:

typedef pair<int, int> pii;
typedef long long LL;

相比 define 之下这样更规范一些,因为 define 是生硬的字符串替换,而 typedef 则是类型别名,能够被编译器检查,所以能使用 typedef 还是使用 typedef。

尾声

除了上面用到的这些头文件之外,还有一些更高端的用法,比如一些模板类,以及一些常用的算法,比如 gcd 等等。但我个人觉得意义不是非常大,对于面试、笔试的代码环节来说,以上的这些头文件已经足够使用了。



☞苹果iCloud(贵安)数据中心投入运行;网易公布不当言论HR处理结果:已开除;OceanBase数据库将开源|极客头条☞程序员35岁就该退休了吗?☞从“不作恶”到“门就在那边”,谷歌 AI 伦理史
Logo

20年前,《新程序员》创刊时,我们的心愿是全面关注程序员成长,中国将拥有新一代世界级的程序员。20年后的今天,我们有了新的使命:助力中国IT技术人成长,成就一亿技术人!

更多推荐