vijos1250 最勇敢的机器人(并查集+分组背包)

用并查集维护连通块,每个连通块一组,跑分组背包。

#include 
#include 
#include 
#include 
using namespace std;
#define N 1005
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,v,m,w[N],c[N],fa[N],f[N];
vector<int> a[N];
inline int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
int main(){
//  freopen("a.in","r",stdin);
    n=read();v=read();m=read();
    for(int i=1;i<=n;++i) w[i]=read(),c[i]=read(),fa[i]=i;
    while(m--){
        int x=read(),y=read();
        int xx=find(x),yy=find(y);
        if(xx!=yy) fa[xx]=yy;
    }
    for(int i=1;i<=n;++i) a[find(i)].push_back(i);
    for(int i=1;i<=n;++i){
        if(a[i].size()==0) continue;
        for(int j=v;j>=0;--j)
            for(int k=0;kif(j-c[a[i][k]]>=0)
                    f[j]=max(f[j],f[j-c[a[i][k]]]+w[a[i][k]]);
    }
    printf("%d\n",f[v]);
    return 0;
}

你可能感兴趣的:(其他oj,并查集)