牛客 14545 B-经商(并查集+01背包)

题目链接:
https://ac.nowcoder.com/acm/problem/14545

题目大意:
小d需要交朋友,但是交一个朋友需要付出精力,然后才能获得利益,只要是在小d的朋友圈中就可以交朋友了,但是可能有些人不在他的朋友圈中。求小d能获得的最大利益是多少?(设小d的下标为1,每个人只能和交际次数只能有一次)

输入:
第一行t,代表测试样例的组数。第二行N,M,C,代表朋友圈一共有N个人,一共有M个朋友圈,C代表小d的精力。
接下来 N-1行,每行输入ai,bi,第i行代表第i个人和第i+1个人交朋友所需要的精力和所获得的的利益。
接下来M行,每行输入x,y,代表x和y可以交朋友。

输出:
一行n,代表最大利益。

大体思路:
利用并查集,将能交朋友的都连通起来,然后再遍历集合,看谁能和1交朋友,如果能和1交朋友,那么在利用01背包,判断是叫这个朋友利益大还是不交这个朋友利益大。

代码:

#include<iostream>
#include<cstring>
#include<cstdio>
#define ms(x) memset(x,0,sizeof(x))
#define maxn 10011
using namespace std;
int fa[maxn];
int a[maxn],b[maxn],c[maxn];//a保存所需精力,b保存所获利益,c代表1的最大收益
void Init(int n)
{//初始化
    for(int i=1;i<=n;i++)
        fa[i]=i;
}

int Find(int x)
{//利用路径压缩查找
    if(fa[x]!=x)
        fa[x]=Find(fa[x]);
    return fa[x];
}

void Union(int x,int y)
{//合并
    x=Find(x);
    y=Find(y);
    if(x!=y)
        fa[x]=y;
}

int main()
{
    int t,N,M,C;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&N,&M,&C);
        Init(N);
        ms(a),ms(b),ms(c);//初始化
        for(int i=2;i<=N;i++)
            scanf("%d%d",&a[i],&b[i]);
        int t1,t2;
        for(int i=1;i<=M;i++)
        {
            scanf("%d%d",&t1,&t2);
            Union(t1,t2);//合并
        }
        for(int i=2;i<=N;i++)
        {//遍历集合
            if(Find(1)==Find(i))//能和1交朋友
            {
                for(int j=C;j>=0;j--)
                {//传说中的01背包
                    if(j>=a[i])
                        c[j]=max(c[j],c[j-a[i]]+b[i]);//拿还是不拿
                }
            }
        }
        printf("%d\n",c[C]);
    }
}

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值