首页 > 代码库 > NYOJ 679 贪婪的商店

NYOJ 679 贪婪的商店

贪婪的商店

时间限制:1000 ms  |  内存限制:65535 KB
难度:3
描述
小明星期天去一家商店买东西,看上了一个玩具,非常想买下来。但是这家商店有这样一个规定,如果要买一件物品A,可能必须要另一件物品B。而要买物品B,可能必须要买另一件物品C。直到买的这件物品不需要买其他物品为止。
经过顾客抗议,商店重新决定如果买一件物品,所需要买其他物品超过一件的话,可以买其中任一件就好。小明钱不多了,他想知道如果要买这件物品,最少要花多少钱。

输入
第一行包含一个整数T(T <= 100).表示测试数据组数。
每组数据第一行包含三个整数N,M,P(1 <= P <= N <= 1000,0 <= m <= 10000),
分别表示商店物品总个数,物品之间关系数量,小明想买物品的编号。
接下来的一行包含N个整数Vi,表示第i个物品的价钱。
接下来的M行每行包含两个整数a,b(1 <= a,b <= N),表示要买物品a就可能要买物品b。
输出
输出“Case #i: ans”(不含引号),i表示第i组数据,ans表示最少花的钱数。
样例输入
2
4 4 1
1 3 2 4
1 2
2 3
2 4
3 4
4 4 2
2 1 3 4
1 2
2 4
1 3
3 4
样例输出
Case #1: 8
Case #2: 5
简单树型DP!
AC码:
#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
using namespace std;
#define MAX 1005
#define INF 9999999
vector<int> adj[MAX];
int dp[MAX],visit[MAX];
int weight[MAX];
int Min(int a,int b)
{
	return a>b?b:a;
}
void DFS(int u)
{
	if(visit[u])
		return;
	int len=adj[u].size(),v;
	if(len==0)
	{
		dp[u]=weight[u];
		return;
	}
	for(int i=0;i<len;i++)
	{
		v=adj[u][i];
		DFS(v);
		visit[v]=1;
		dp[u]=Min(dp[u],dp[v]+weight[u]);
	}
}
int main()
{
	int T,i,count=1;
	int n,m,p,a,b;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d",&n,&m,&p);
		for(i=0;i<=n;i++)
			adj[i].clear();
		for(i=1;i<=n;i++)
			scanf("%d",&weight[i]);
		for(i=1;i<=m;i++)
		{
			scanf("%d%d",&a,&b);
			adj[a].push_back(b);
		}
		memset(visit,0,sizeof(visit));
		for(i=0;i<=n;i++)
			dp[i]=INF;
		DFS(p);
		printf("Case #%d: %d\n",count++,dp[p]);
	}
	return 0;
}