[STL乱搞]51 Nod——1573 美丽的集合

[STL乱搞]51 Nod——1573 美丽的集合

题目梗概

在多重集之中，同一个元素可以出现多次。

我们现在有n个多重集合，第i个集合最开始都有一个元素ai(1≤i≤n)。

定义多重集合的价值为子集中不同的和的个数。

现在我们有两种操作：

1、合并最开始时多重集合i现在所在的多重集合与多重集合j现在所在的多重集合，成为一个新的多重集合。

2、询问最开始多重集合i现在所在多重集合的价值。

$n<=1000$ $Σai<=100000$ $Q<=2000$

解题思路

集合数和操作数，以及背包大小都不大，直接按题目启发式合并就可以了。

以下是关于题目的一些技巧：

1.用bitset存每个集合的价值，1表示存在这个和，合并集合时直接位运算就可以了。

2.用vector存每个集合的元素。

所以是一道STl乱搞题。

#include<cstdio>
#include<bitset>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=1005;
bitset<100005> a[maxn];
vector<int> h[maxn];
int n,Q,x,y,d,f[maxn];
int get(int x){
    if (f[x]==x) return x;
    f[x]=get(f[x]);
    return f[x];
}
void merge(int x,int y){
    if (h[x].size()>h[y].size()) swap(x,y);
    for (int i=0;i<h[x].size();i++) h[y].push_back(h[x][i]),a[y]=a[y]|(a[y]<<h[x][i]);
    f[x]=y;
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    scanf("%d",&n);
    for (int i=1;i<=n;i++){
        scanf("%d",&x);
        f[i]=i;a[i][0]=a[i][x]=1;h[i].push_back(x);
    }
    scanf("%d",&Q);
    while(Q--){
        scanf("%d%d",&d,&x);x=get(x);
        if (d==1){
            scanf("%d",&y);y=get(y);
            merge(x,y);
        }else printf("%d\n",a[x].count());
    }
}