hdu 2415 Bribing FIPA

樹形DP,不同的是,只要選擇一個根節點i,它的子樹中的所有節點也被選擇,代價爲cost[i]

由於dfs時,先遍歷子樹,再返回根節點,這樣,以i爲根的子樹中的節點可能被多次利用,解決辦法就是訪問到i時,先開一個tmp數組當前的狀態,即dp值,然後訪問子樹,返回時,先更新tmp值,再用tmp值更新dp值

還有一個trick就是,題目要求的是至少m張選票,不是剛好m張,所以可以大於m

代碼:

#include<iostream>
#include<memory.h>
#include<string>
#include<cstdio>
#include<algorithm>
#include<math.h>
#include<stack>
#include<queue>
#include<vector>
#include<map>
using namespace std;
struct node 
{
	int v,next;
}g[205*205];
map<string,int>mp;
int adj[205],dp[405],cnt[205],e,in[205],cost[205];
int n,m;
char s[105];
void add(int u,int v)
{
	g[e].v=v; g[e].next=adj[u]; adj[u]=e++;
}
void dfs(int u)
{
	int i,k,v;
	cnt[u]=1;
	int tmp[405];
	memcpy(tmp,dp,sizeof(dp));
	for(i=adj[u];i!=-1;i=g[i].next)
	{
		v=g[i].v;
		dfs(v);
		cnt[u]+=cnt[v];
	}
	for(i=m+cnt[u];i>=cnt[u];i--)
	{
		//k=min(i,m);
		tmp[i]=min(tmp[i],tmp[i-cnt[u]]+cost[u]);
		dp[i]=min(dp[i],tmp[i]);
	}
}
int main()
{
	int i,j,k,l,w,idx;
	char c;
	while(gets(s))
	{
		if(s[0]=='#')
			break;
		sscanf(s,"%d%d",&n,&m);
		mp.clear();
		memset(adj,-1,sizeof(adj));
		memset(in,0,sizeof(in));
		e=idx=0;
		for(i=1;i<=n;i++)
		{
			dp[i]=dp[i+n]=1<<20;
			scanf("%s%d",s,&w);
			k=mp[s];
			if(k==0)
			{
				mp[s]=++idx;
				k=idx;
			}
			cost[k]=w;
			c=getchar();
			while(c==' ')
			{
				scanf("%s",s);
				l=mp[s];
				if(l==0)
				{
					mp[s]=++idx;
					l=idx;
				}
				add(k,l);
				in[l]++;
				c=getchar();
			}
		}
		dp[0]=0;
		for(i=1;i<=n;i++)
		{
			if(in[i]==0)
				dfs(i);
		}
		int ans=1<<20;
		for(i=m;i<=2*n;i++)
			ans=min(dp[i],ans);
		printf("%d\n",ans);
	}
	return 0;
}


發佈了242 篇原創文章 · 獲贊 5 · 訪問量 23萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章