P4306 [JSOI2010]连通数
做强连通分量的时候刷到的题,看到题解惊呆了。太太太秀了。
#include <bits/stdc++.h>
#define N 2010
#define LL long long
using namespace std;
char s[N];
bitset<N>f[N];
int main()
{
int n,p,i,a,j;
LL sum=0;
scanf("%d",&n);
for(i=1;i<=n;++i)
{
scanf("\n%s",s+1);
for(j=1;j<=n;++j)
if(s[j]=='1') f[i][j]=1;
f[i][i]=1;
}
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
if(f[j][i]) f[j]|=f[i];
for(i=1;i<=n;++i) sum+=f[i].count();
printf("%lld",sum);
}
bitset存储二进制数位。
bitset就像一个bool类型的数组一样,但是有空间优化——bitset中的一个元素一般只占1 bit,相当于一个char元素所占空间的八分之一。
bitset中的每个元素都能单独被访问,例如对于一个叫做foo的bitset,表达式foo[3]访问了它的第4个元素,就像数组一样。
bitset有一个特性:整数类型和布尔数组都能转化成bitset。
bitset的大小在编译时就需要确定。如果你想要不确定长度的bitset,请使用(奇葩的)vector<bool>。
bitset的运算
bitset的运算就像一个普通的整数一样,可以进行与(&)、或(|)、异或(^)、左移(<<)、右移(>>)等操作。
bitset的相关函数
对于一个叫做foo的bitset:foo.size()
返回大小(位数)foo.count()
返回1的个数foo.any()
返回是否有1foo.none()
返回是否没有1foo.set()
全都变成1foo.set(p)
将第p + 1位变成1foo.set(p, x)
将第p + 1位变成xfoo.reset()
全都变成0foo.reset(p)
将第p + 1位变成0foo.flip()
全都取反foo.flip(p)
将第p + 1位取反foo.to_ulong()
返回它转换为unsigned long的结果,如果超出范围则报错foo.to_ullong()
返回它转换为unsigned long long的结果,如果超出范围则报错foo.to_string()
返回它转换为string的结果