1252. 搭配购买 并查集 + 01背包

题目

题解思路

利用并查集将搭配合并起来 。
将每个连通块 进行 01 背包操作 这样复杂度最大 m*w 再对01背包进行一维优化 即可。

AC代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <vector>
#include <algorithm>
#include <map>
#include <string>
using namespace std;

const  int  INF =  0x3f3f3f3f;

struct node
{
    int p,c,d;
}a[10010];


long long dp[5010];

int find2(int x )
{
    int r = x;
    while( a[r].p != r )
    {
        r = a[r].p;
    }
    while( x != a[x].p )
    {
        int t = a[x].p ;
        a[x].p = r ;
        x = t ;
    }

    return r;
}


bool uio(int x , int y )
{
    int fx = find2(x);
    int fy = find2(y);
    if ( fx != fy )
    {
        a[fy].p = fx ;
        a[fx].c += a[fy].c ;
        a[fx].d += a[fy].d ;
        return 1;
    }
    return 0 ;
}

int main ()
{
    ios::sync_with_stdio(false);
    int n,m,sum;
    cin>>n>>m>>sum;
    for (int i = 1 ; i <= n ; i++ )
    {
        int t1,t2;
        cin>>t1>>t2;
        a[i].p = i ;
        a[i].c = t1 ;
        a[i].d = t2 ;
    }
    for (int i = 1 ; i <= m ; i++ )
    {
        int t1,t2;
        cin>>t1>>t2;
        uio(t1,t2);
    }
    for (int i = 1 ; i <= n ; i++ )
        if ( a[i].p == i )
            for (int j = sum ; j >=  a[i].c ; j-- )
                dp[j] = max( dp[j] , dp[j - a[i].c ] + a[ i ].d );
    cout<<dp[sum]<<"\n";
    return 0 ;
}