51nod 1573 美丽的集合(bitset)

来源:互联网 发布:一号防线监控网络设置 编辑:程序博客网 时间:2024/06/06 00:51

正如题解所说,这个可以暴力,但是必须用bitset,因为这个是lg(n)的算法。所以我直接暴力错了几组测试数据拿了5分。。。心痛,只用过一次bitset,然而当时想到了这个,并不熟练也不知道他的复杂度怎么来的。

#include<cstdio>#include<bitset>#include <iostream>using namespace std;const int N=1005;bitset<100001> a[N],b;int n,i,f[N],g[N],d[N][N];int read(){char c=getchar();int k=0;for (;c<48||c>57;c=getchar());for (;c>47&&c<58;c=getchar()) k=(k<<3)+(k<<1)+c-48;return k;}int get(int x){return f[x]==x?x:f[x]=get(f[x]);}int main(){for (n=read(),i=1;i<=n;i++)a[i][0]=a[i][d[i][g[i]=1]=read()]=1,f[i]=i;//把a[i][0]和a[i][now]赋值为1,now是刚才输入的for (int Q=read();Q--;){if (read()==2){printf("%d\n",a[get(read())].count());continue;}int x=get(read()),y=get(read());if (g[y]>g[x]) swap(x,y);f[y]=x;for (i=g[y];i;i--)    a[x]|=a[x]<<(d[x][++g[x]]=d[y][i]);//这个是并的操作,开始我还以为是亦或操作,亦或见多了迷了,这个就是所有的和,但是时间复杂度低}}