51nod - 1573 美丽的集合

 

多重集合是数学中的一个概念，是集合概念的推广。在一个集合中，相同的元素只能出现一次，因此只能显示出有或无的属性。在多重集之中，同一个元素可以出现多次。

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

一个拥有n个元素的多重集合显然会有2^n个子集。

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

例如一个多重集合{1,2}，那么会存在有4个不同的和{0,1,2,3}。其价值为4。

一个多重集合{2,2}，那么会有3个不同的和{0,2,4},其价值为3。

现在我们有两种操作：

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

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

   数据保证合并两个最开始时的多重集合所在的多重集合时它们当前不是同一个多重集合。

 

样例解释：

第一个查询的多重集合为{1}，价值为2

第二个查询的多重集合为{1,1}，价值为3

第三个查询的多重集合为{2,4}，价值为4

 

Input

第一行一个数n，表示有n个多重集合。(1<=n<=1000)
第二行n个数ai。（Σai<=100000，ai>0）
第三行一个数Q，表示操作个数。(1<=Q<=2000)
接下来Q行，一开始读入一个数A，表示操作A。
若A为1，接下来读入两个数i和j，表示合并操作(1<=i,j<=n)。
若A为2，接下来读入一个数i，表示查询操作(1<=i<=n)。

Output

对于所有A=2的操作，输出相应的答案。

Input示例

5
1 1 2 4 5
5
2 1
1 1 2
2 2
1 3 4
2 3

Output示例

2
3
4

思路：

利用并查集来记录每次合并的结果。利用bitset来统计组合的数量，比如第i的位置有一个数3，则bitset值为1001，即有0和3两种组合，若向第i个位置再添加一个3，则首先分别计算原来的组合加上3以后的值，则将bitset值左移三位即可，即变为1001000，就是说加上3以后原来的组合变成3和6。这时再将1001000与原来的bitset值1001进行或操作得到1001001，这样就可得到所有的组合0，3和6。

#include <iostream>
#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 getFather(int x)
{
    if (f[x] == x)
	{
		return x;
	}
    f[x] = getFather(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] << h[x][i]);
	}
 
    f[x] = y;
}
 
int main()
{
	cin >> n;
    for (int i = 1; i <= n; i++)
	{
		cin >> x;
        f[i] = i;
		a[i][0] = a[i][x] = 1;
		h[i].push_back(x);
    }
	cin >> Q;
    while (Q--)
	{
		cin >> d >> x;
		x = getFather(x);
        if (d == 1)
		{
			cin >> y;
			y = getFather(y);
            merge(x, y);
        }
		else
		{
			cout << a[x].count() << endl;
		}
    }
 
	return 0;
}