ZCMU1384 简单题4(搜索)

Description

ACboy很喜欢玩一种战略游戏，在一个地图上，有N座城堡，每座城堡都有一定的宝物，在每次游戏中ACboy允许攻克M个城堡并获得里面的宝物。但由于 地理位置原因，有些城堡不能直接攻克，要攻克这些城堡必须先攻克其他某一个特定的城堡。你能帮ACboy算出要获得尽量多的宝物应该攻克哪M个城堡吗？

Input

每个测试实例首先包括2个整数，N,M.(1 <= M <= N <= 20);在接下来的N行里，每行包括2个整数，a,b. 在第 i 行，a 代表要攻克第 i 个城堡必须先攻克第 a 个城堡，如果 a = 0 则代表可以直接攻克第 i 个城堡。b 代表第 i 个城堡的宝物数量, b >= 0。当N = 0, M = 0输入结束。

Output

对于每个测试实例，输出一个整数，代表ACboy攻克M个城堡所获得的最多宝物的数量。

Sample Input

3 2

0 1

0 2

0 3

7 4

2 2

0 1

0 4

2 1

7 1

7 6

2 2

0 0

Sample Output

5

13

代码

#include<stdio.h>
#include<algorithm>
#define MAX 50
using namespace std;
int c[MAX][MAX],mark[MAX],tre[MAX],n,m,maxl;
void init()
{
    for(int i=0;i<=n;i++)
    {
        mark[i]=tre[i]=0;
        for(int j=0;j<=n;j++)
        {
             c[i][j]=-1;
        }
    }
}
void DFS(int start,int v,int step)
{
    if(step==m)
    {
        if(v>maxl) maxl=v;
        return ;
    }
    for(int i=0;i<=n;i++)
    {
        if(c[start][i]==1&&!mark[i])
        {
            mark[i]=1;
            DFS(start,v+tre[i],step+1);
            DFS(i,v+tre[i],step+1);
            mark[i]=0;
        }
        else if(c[start][i]==2&&!mark[i])
        {
            mark[i]=1;
            DFS(i,v+tre[i],step+1);
            mark[i]=0;
        }
    }
    return ;
}
int main()
{
    int u,val;
    while(~scanf("%d%d",&n,&m))
    {
        if(!n&&!m) break;
        init();
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&u,&val);
            if(u==0)
                c[u][i]=1;
            else
                c[u][i]=2;
            tre[i]=max(val,tre[i]);
        }
        maxl=-1;
        for(int i=0;i<=n;i++)
        {
            if(i==0)
                DFS(i,0,0);
            else
            {
               mark[i]=1;
               DFS(i,0,1);
               mark[i]=0;
            }
        }
        printf("%d\n",maxl);
    }
    return 0;
}