hdu 4284(状压dp)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4284

思路：类似于poj3311:http://poj.org/problem?id=3311,首先floyd预处理出两点之间的最短距离，然后就是枚举所有的状态了。

 1 #include<iostream>

 2 #include<cstdio>

 3 #include<cstring>

 4 #include<algorithm>

 5 using namespace std;

 6 #define MAXN 111

 7 #define inf 0x3f3f3f3f

 8 

 9 int map[MAXN][MAXN];

10 int dp[1<<17][17];

11 int city[17],cost[17],value[17];

12 int N,M,H,money;

13 

14 void floyd()

15 {

16     for(int k=1;k<=N;k++)

17         for(int i=1;i<=N;i++)

18             for(int j=1;j<=N;j++)

19                 if(map[i][k]<inf&&map[k][j]<inf&&map[i][k]+map[k][j]<map[i][j])

20                     map[i][j]=map[i][k]+map[k][j];

21 }

22 

23 

24 int main()

25 {

26     int _case,u,v,w;

27     scanf("%d",&_case);

28     while(_case--){

29         scanf("%d%d%d",&N,&M,&money);

30         for(int i=1;i<=N;i++)

31             for(int j=1;j<=N;j++)

32                 map[i][j]=(i==j)?0:inf;

33         while(M--){

34             scanf("%d%d%d",&u,&v,&w);

35             map[u][v]=map[v][u]=min(map[u][v],w);

36         }

37         floyd();

38         scanf("%d",&H);

39         for(int i=0;i<H;i++){

40             scanf("%d%d%d",&city[i],&value[i],&cost[i]);

41         }

42         memset(dp,-1,sizeof(dp));

43         for(int i=0;i<H;i++){

44             int tmp=money-map[1][city[i]]-cost[i];

45             if(tmp>=0)dp[(1<<i)][i]=tmp+value[i];

46         }

47         for(int state=0;state<(1<<H);state++){

48             for(int i=0;i<H;i++){

49                 if((state&(1<<i))==0)continue;

50                 if(dp[state][i]<0)continue;

51                 for(int j=0;j<H;j++){

52                     if(state&(1<<j))continue;

53                     if(map[city[i]][city[j]]==inf)continue;

54                     if(dp[state][i]-map[city[i]][city[j]]-cost[j]<0)continue;

55                     dp[state|(1<<j)][j]=max(dp[state|(1<<j)][j],dp[state][i]-map[city[i]][city[j]]-cost[j]+value[j]);

56                 }

57             }

58         }

59         bool flag=false;

60         for(int i=0;i<H;i++){

61             if(dp[(1<<H)-1][i]-map[city[i]][1]>=0){

62                 flag=true;

63                 break;

64             }

65         }

66         flag?puts("YES"):puts("NO");

67     }

68     return 0;

69 }

70 

71             

72 

73         

74         

75 

76 

77 

78         

View Code