hdu 3001(状压dp）

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

思路：这道题类似于TSP问题，只不过题目中说明每个城市至少要走一次，至多走2次，因此要用到三进制压缩，然后就是状态转移方程了：dp[i+State[k]]=min(dp[i+State[k]],dp[i][j]+map[j][k]);

 1 #include<iostream>

 2 #include<cstdio>

 3 #include<cstring>

 4 #include<algorithm>

 5 using namespace std;

 6 #define inf 0x3f3f3f3f

 7 int State[12];

 8 int visited[60000][12];

 9 int dp[60000][12];

10 int map[12][12];

11 int n,m;

12 

13 void Initiate()

14 {

15     State[0]=1;

16     for(int i=1;i<=10;i++)

17         State[i]=State[i-1]*3;

18     for(int i=0;i<=State[10];i++){

19         int tmp=i;

20         for(int j=0;j<=10;j++){

21             visited[i][j]=tmp%3;

22             tmp/=3;

23         }

24     }

25 }

26 

27 int main()

28 {

29     Initiate();

30     int u,v,w,ans;

31     while(~scanf("%d%d",&n,&m)){

32         memset(dp,inf,sizeof(dp));

33         memset(map,inf,sizeof(map));

34         for(int i=0;i<n;i++)dp[State[i]][i]=0;

35         while(m--){

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

37             u--,v--;

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

39         }

40         ans=inf;

41         for(int i=0;i<State[n];i++){

42             bool flag=true;

43             for(int j=0;j<n;j++){

44                 if(visited[i][j]==0)flag=false;//是否每个城市都至少走了1次

45                 if(dp[i][j]==inf)continue;

46                 for(int k=0;k<n;k++)if(j!=k){

47                     if(visited[i][k]>=2)continue;

48                     if(map[j][k]==inf)continue;

49                     dp[i+State[k]][k]=min(dp[i+State[k]][k],dp[i][j]+map[j][k]);

50                 }

51             }

52             if(flag){

53                 for(int j=0;j<n;j++){

54                     ans=min(ans,dp[i][j]);

55                 }

56             }

57         }

58         if(ans==inf)ans=-1;

59         printf("%d\n",ans);

60     }

61     return 0;

62 }

63 

64 

65 

66 

67 

68     

View Code