这道题考察了并查集和背包(一开始写的时候还不会用背包,用深搜代替了背包的部分然后一半的检查点时间超限了,果然专业的事就得用专业的工具)
原题点这里-->P1455 搭配购买 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
复制Markdown 展开
题目描述
明天就是母亲节了,电脑组的小朋友们在忙碌的课业之余挖空心思想着该送什么礼物来表达自己的心意呢?听说在某个网站上有卖云朵的,小朋友们决定一同前往去看看这种神奇的商品,这个店里有 n 朵云,云朵已经被老板编号为 1,2,3,...,n,并且每朵云都有一个价值,但是商店的老板是个很奇怪的人,他会告诉你一些云朵要搭配起来买才卖,也就是说买一朵云则与这朵云有搭配的云都要买,电脑组的你觉得这礼物实在是太新奇了,但是你的钱是有限的,所以你肯定是想用现有的钱买到尽量多价值的云。
输入格式
第一行输入三个整数,n,m,w,表示有 n 朵云,m 个搭配和你现有的钱的数目。
第二行至 n+1 行,每行有两个整数,ci,di,表示第 i 朵云的价钱和价值。
第 n+2 至 n+1+m 行 ,每行有两个整数 ui,vi。表示买第 ui 朵云就必须买第 vi 朵云,同理,如果买第vi 朵就必须买第ui 朵。
输出格式
一行,表示可以获得的最大价值。
输入输出样例
输入 #1
5 3 10 3 10 3 10 3 10 5 100 10 1 1 3 3 2 4 2
输出 #1
1
说明/提示
- 对于30% 的数据,满足 1≤n≤100;
- 对于50% 的数据,满足 1≤n,w≤1000,1≤m≤100;
- 对于100% 的数据,满足 1≤n,w≤10000,0≤m≤5×1000。
解题思路
首先我们用并查集把所有需要搭配的云组合成一团云,并把每团云的总价格和总价值记录在他们各自的根节点上。
#include<stdio.h>
#define MAX 10001
typedef struct{
int price;
int value;
int follow;
}cloud;
cloud s[MAX];
int find(int x)//查
{
while(s[x].follow>0)
{
x=s[x].follow;
}
return x;
}
void unite(int x,int y)//并
{
int root_x=find(x),root_y=find(y);
if(root_x==root_y)
return;
if(s[root_x].follow>s[root_y].follow)
{
s[root_y].follow+=s[root_x].follow;
s[root_y].price+=s[root_x].price;
s[root_y].value+=s[root_x].value;
s[root_x].follow=root_y;
}
else
{
s[root_x].follow+=s[root_y].follow;
s[root_x].price+=s[root_y].price;
s[root_x].value+=s[root_y].value;
s[root_y].follow=root_x;
}
}
int max(int a,int b)
{
return (a>b)?a:b;
}
int f[10003],k=1;
int main()
{
int n,m,W;
scanf("%d%d%d",&n,&m,&W);
for(int i=1;i<=n;i++)
{
int c,d;
scanf("%d%d",&c,&d);
s[i].follow=-1;
s[i].price=c;
s[i].value=d;
}
for(int i=0;i<m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
unite(x,y);
}
然后把每个根节点的价格和价值用对应的数组记录起来为01背包做准备
int w[10002],v[10002];//分别用来记录每团云的价格和价值
for(int i=1;i<=n;i++)//遍历整个并查集
{
if(s[i].follow<0&&s[i].price<=W)//如果价格大于手上的钱就不用记录
{
w[k]=s[i].price;
v[k++]=s[i].value;
}
}
最后交给01背包求最优解
for(int i=1;i<k;i++)
{
for(int j=W;j>=w[i];j--)
{
f[j]=max(f[j],f[j-w[i]]+v[i]);
}
}
源代码
#include<stdio.h>
#define MAX 10001
typedef struct{
int price;
int value;
int follow;
}cloud;
cloud s[MAX];
int find(int x)
{
while(s[x].follow>0)
{
x=s[x].follow;
}
return x;
}
void unite(int x,int y)
{
int root_x=find(x),root_y=find(y);
if(root_x==root_y)
return;
if(s[root_x].follow>s[root_y].follow)
{
s[root_y].follow+=s[root_x].follow;
s[root_y].price+=s[root_x].price;
s[root_y].value+=s[root_x].value;
s[root_x].follow=root_y;
}
else
{
s[root_x].follow+=s[root_y].follow;
s[root_x].price+=s[root_y].price;
s[root_x].value+=s[root_y].value;
s[root_y].follow=root_x;
}
}
int max(int a,int b)
{
return (a>b)?a:b;
}
int f[10003],k=1;
int main()
{
int n,m,W;
scanf("%d%d%d",&n,&m,&W);
for(int i=1;i<=n;i++)
{
int c,d;
scanf("%d%d",&c,&d);
s[i].follow=-1;
s[i].price=c;
s[i].value=d;
}
for(int i=0;i<m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
unite(x,y);
}
int w[10002],v[10002];
for(int i=1;i<=n;i++)
{
if(s[i].follow<0&&s[i].price<=W)
{
w[k]=s[i].price;
v[k++]=s[i].value;
}
}
for(int i=1;i<k;i++)
{
for(int j=W;j>=w[i];j--)
{
f[j]=max(f[j],f[j-w[i]]+v[i]);
}
}
printf("%d",f[W]);
return 0;
}
01背包是为了写这道题才学的,第一次写的01背包用的是最容易理解的二维数组的形式,当数据量拉到题目最大时,我的二维数组里的后面一小部分的数据全是垃圾值,显然是因为用二维数组空间消耗太大导致数据溢出了,最后又去学了一维数组形式的01背包才AC了这道题。