能量获取
(energy.pas/c/cpp)
题目描述
“封印大典启动,请出 Nescafe 魂珠!”随着圣主 applepi 一声令下,圣剑护法 rainbow和魔杖护法 freda 将 Nescafe 魂珠放置于封印台上。封印台是一个树形的结构,魂珠放置的位置就是根节点(编号为 0)。还有 n 个其它节点(编号 1~n)上放置着封印石,编号为 i的封印石需要从魂珠上获取 Ei 的能量。能量只能沿着树边从魂珠传向封印石,每条边有一个能够传递的能量上限 Wi,魂珠的能量是无穷大的。作为封印开始前的准备工作,请你求出最多能满足多少颗封印石的能量需求?
注意:能量可以经过一个节点,不满足它的需求而传向下一个节点。每条边仅能传递一
次能量。
输入格式
第一行一个整数 n,表示除根节点之外其它节点的数量。
接下来 n 行,第 i+1 行有三个整数 Fi、Ei、Wi,分别表示 i 号节点的父节点、i 号节点上封印石的能量需求、连接节点 i 与 Fi 的边最多能传递多少能量。
输出格式
最多能满足多少颗封印石的能量需求。
样例输入
4
0 3 2
0 100 100
1 1 1
2 75 80
样例输出
2
数据范围与约定
对于 100% 的数据,满足 1<=n<=1000,0<=Fi<=n,0<=Ei,Wi<=100。
考试时想到了用贪心,但是证明了半天仍然不能确定其正确性,结果……送分题只得了20分。血的教训(雾)告诉我们考试的话就不要分时间证明正确性了,敢写才有可能得分。
正解思路:
贪心,每次选取能量需求最小的节点,扫描它到根节点的路径上的边的容量,看能否满足,如果能满足就把它到根节点的路径上的边的容量都减去它的需求即可。
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int MAXN=1000,INF=1001;
int n;
struct dat{int value,id;}en[MAXN+1];
int head[MAXN+1],ecnt;
struct node{int next,to,value;} e[MAXN+1];
void add(int x,int y,int z)
{
ecnt++;
e[ecnt].to=y;
e[ecnt].next=head[x];
head[x]=ecnt;
e[ecnt].value=z;
}
bool cmp(const dat & a,const dat & b){return a.value<b.value;}
int ans=0;
int main()
{
freopen("energy.in","r",stdin);
freopen("energy.out","w",stdout);
int i;
scanf("%d",&n);
int F,E,W;
for(i=1;i<=n;i++)
{
scanf("%d%d%d",&F,&E,&W);
add(i,F,W);
en[i].value=E;
en[i].id=i;
}
sort(en+1,en+n+1,cmp);
int minv=INF,tmp;
for(i=1;i<=n;i++)
{
minv=INF;
for(tmp=en[i].id;tmp;tmp=e[head[tmp]].to)
if(e[head[tmp]].value<minv)
minv=e[head[tmp]].value;
if(minv>=en[i].value)
{
ans++;
for(tmp=en[i].id;tmp;tmp=e[head[tmp]].to)
e[head[tmp]].value-=en[i].value;
}
}
printf("%d\n",ans);
return 0;
}