正如题解所说,这个可以暴力,但是必须用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]);//这个是并的操作,开始我还以为是亦或操作,亦或见多了迷了,这个就是所有的和,但是时间复杂度低
}
}