1573 美丽的集合-------------------思维(启发式合并+bitset)

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述
解析:
利用启发式合并 nlogn
然后利用bitset来维护集合中不同和的次数
合并的时候用链表连接同一集合的元素
用en[]更新每个集合的最后一个元素是谁

#include<bits/stdc++.h>
using namespace std;
const int N=1005;
bitset<100005>f[N];
int fa[N],size[N],nx[N],a[N],en[N];
int n,m;
int find(int x)
{
  if(fa[x]!=x) return fa[x]=find(fa[x]);
  return fa[x];
}
void unio(int u,int v)
{
  int x=find(u);
  int y=find(v);
  if(size[x]>size[y])swap(x,y);

  for(int i=x;i;i=nx[i])
  {
    f[y]|=(f[y]<<a[i]);
  }
//  cout<<y<<"   "<<f[y]<<endl;
  nx[en[y]]=x;
  en[y]=en[x];
  fa[x]=y;size[y]+=size[x];
}
int main()
{
  scanf("%d",&n);
  for(int i=0;i<N;i++) fa[i]=i;
  for(int i=1;i<=n;i++) scanf("%d",&a[i]),size[i]=1,en[i]=i,f[i][a[i]]=f[i][0]=1;
  scanf("%d",&m);
  while(m--)
  {
    int op;
    scanf("%d",&op);
    if(op==1){
      int x,y;
      scanf("%d %d",&x,&y);
      unio(x,y);
    }
    else
    {
      int x;
      scanf("%d",&x);
      int u=find(x);
    //  cout<<u<<endl;
      printf("%d\n",f[u].count());
    }
  }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值