现在的位置: 首页 > 综合 > 正文

POJ3345 Bribing FIPA

2018年05月02日 ⁄ 综合 ⁄ 共 2147字 ⁄ 字号 评论关闭

树形DP,不同的是,只要选择一个根节点i,它的子树中的所有节点也被选择,代价为cost[i]

由于dfs时,先遍历子树,再返回根节点,这样,以i为根的子树中的节点可能被多次利用,解决办法就是访问到i时,先开一个tmp数组当前的状态,即dp值,然后访问子树,返回时,先更新tmp值,再用tmp值更新dp值

还有一个trick就是,题目要求的是至少m张选票,不是刚好m张,所以可以大于m

代码:

  1. #include<iostream>  
  2. #include<memory.h>  
  3. #include<string>  
  4. #include<cstdio>  
  5. #include<algorithm>  
  6. #include<math.h>  
  7. #include<stack>  
  8. #include<queue>  
  9. #include<vector>  
  10. #include<map>  
  11. using namespace std;  
  12. struct node   
  13. {  
  14.     int v,next;  
  15. }g[205*205];  
  16. map<string,int>mp;  
  17. int adj[205],dp[405],cnt[205],e,in[205],cost[205];  
  18. int n,m;  
  19. char s[105];  
  20. void add(int u,int v)  
  21. {  
  22.     g[e].v=v; g[e].next=adj[u]; adj[u]=e++;  
  23. }  
  24. void dfs(int u)  
  25. {  
  26.     int i,k,v;  
  27.     cnt[u]=1;  
  28.     int tmp[405];  
  29.     memcpy(tmp,dp,sizeof(dp));  
  30.     for(i=adj[u];i!=-1;i=g[i].next)  
  31.     {  
  32.         v=g[i].v;  
  33.         dfs(v);  
  34.         cnt[u]+=cnt[v];  
  35.     }  
  36.     for(i=m+cnt[u];i>=cnt[u];i--)  
  37.     {  
  38.         //k=min(i,m);  
  39.         tmp[i]=min(tmp[i],tmp[i-cnt[u]]+cost[u]);  
  40.         dp[i]=min(dp[i],tmp[i]);  
  41.     }  
  42. }  
  43. int main()  
  44. {  
  45.     int i,j,k,l,w,idx;  
  46.     char c;  
  47.     while(gets(s))  
  48.     {  
  49.         if(s[0]=='#')  
  50.             break;  
  51.         sscanf(s,"%d%d",&n,&m);  
  52.         mp.clear();  
  53.         memset(adj,-1,sizeof(adj));  
  54.         memset(in,0,sizeof(in));  
  55.         e=idx=0;  
  56.         for(i=1;i<=n;i++)  
  57.         {  
  58.             dp[i]=dp[i+n]=1<<20;  
  59.             scanf("%s%d",s,&w);  
  60.             k=mp[s];  
  61.             if(k==0)  
  62.             {  
  63.                 mp[s]=++idx;  
  64.                 k=idx;  
  65.             }  
  66.             cost[k]=w;  
  67.             c=getchar();  
  68.             while(c==' ')  
  69.             {  
  70.                 scanf("%s",s);  
  71.                 l=mp[s];  
  72.                 if(l==0)  
  73.                 {  
  74.                     mp[s]=++idx;  
  75.                     l=idx;  
  76.                 }  
  77.                 add(k,l);  
  78.                 in[l]++;  
  79.                 c=getchar();  
  80.             }  
  81.         }  
  82.         dp[0]=0;  
  83.         for(i=1;i<=n;i++)  
  84.         {  
  85.             if(in[i]==0)  
  86.                 dfs(i);  
  87.         }  
  88.         int ans=1<<20;  
  89.         for(i=m;i<=2*n;i++)  
  90.             ans=min(dp[i],ans);  
  91.         printf("%d\n",ans);  
  92.     }  
  93.     return 0;  
  94. }  

抱歉!评论已关闭.