HDU1258 DFS

题意：求n个数中的某些数的和等于t，并输出

dfs

记录下已经输出过的，然后每次比较一下，这样就能避免重复

View Code

  1 #include<stdio.h>
  2 #include<string.h>
  3 #include<stdlib.h>
  4 #include<algorithm>
  5 using namespace std;
  6 const int maxn = 15;
  7 const int maxm = 500005;
  8 int num[ maxn ];
  9 int vis[ maxn ];
 10 int ans[ maxn ];
 11 int t,n;
 12 int flag;
 13 int new_start;
 14 struct node{
 15     int cnt;
 16     int ans[ maxn ];
 17 }my[ maxm ],temp;
 18 int cnt_my;
 19 int judge( ){
 20     sort( temp.ans,temp.ans+temp.cnt );
 21     for( int i=0;i<cnt_my;i++ ){
 22         if( temp.cnt==my[i].cnt ){
 23             int k;
 24             int fff=1;
 25             for( k=0;k<temp.cnt;k++ ){
 26                 if( temp.ans[k]==my[i].ans[k] )
 27                 {}
 28                 else{
 29                     fff=-1;
 30                     break;
 31                 }
 32             }
 33             if( fff==1 )
 34                 return i;//exist
 35         }
 36     }
 37     for( int i=0;i<temp.cnt;i++ ){
 38         my[ cnt_my ].ans[ i ]=temp.ans[i];
 39     }
 40     my[ cnt_my ].cnt=temp.cnt;
 41     cnt_my++;
 42     return -1;
 43 }
 44 int judge2(int p){
 45     int sum=0;
 46     for( int i=p;i<=n;i++ ){
 47         sum+=num[i];
 48     }
 49     return sum;
 50 }
 51 void dfs( int pos,int sum,int cnt ){
 52     if( sum==t ){
 53         temp.cnt=cnt;
 54         for( int i=0;i<cnt;i++ ){
 55             temp.ans[i]=ans[i];
 56         }
 57         if( judge()==-1 ){
 58             flag=1;
 59             for( int i=0;i<cnt;i++ ){
 60                 if( i==0 ) printf("%d",ans[i]);
 61                 else printf("+%d",ans[i]);
 62             }
 63             printf("\n");
 64         }
 65         //new_start++;
 66         return ;
 67     }
 68     for( int i=new_start;i<=n;i++ ){
 69         if( vis[ i ]==0&&(sum+num[i])<=t ){
 70             int ttt=judge2( i );
 71             for( int k=0;k<cnt;k++ )
 72                 ttt+=ans[k];
 73             if( ttt<t ) return ;
 74             vis[ i ]=1;
 75             ans[ cnt ]=num[i];
 76             dfs( i,sum+num[i],cnt+1 );
 77             vis[ i ]=0;
 78         }
 79     }
 80     return ;
 81 }
 82 
 83 int main(){
 84     while( scanf("%d%d",&t,&n),n ){
 85         for( int i=1;i<=n;i++ ){
 86             scanf("%d",&num[ i ]);
 87             vis[i]=0;
 88         }
 89         flag=-1;
 90         cnt_my=0;
 91         printf("Sums of %d:\n",t);
 92         for( int i=1;i<=n;i++ ){
 93             if( num[i]>t ) {
 94                 vis[i]=1;
 95                 continue;
 96             }
 97             if( num[i]==t&&flag==-1 ){//在这个判断条件上wa了一次！！！flag！=-1
 98                 flag=1;
 99                 vis[i]=1;
100                 my[ 0 ].cnt=1;
101                 my[ 0 ].ans[0]=t;
102                 cnt_my++;
103                 printf("%d\n",t);
104                 continue;
105             }
106             if( num[i]==t&&flag==1 ){
107                 vis[i]=1;
108                 continue;
109             }
110             new_start=i;
111             dfs( i,0,0 );//pos sum cnt
112             break;
113         }
114         if( flag==-1 )
115             printf("NONE\n");
116         //printf("end\n");
117     }
118     return 0;
119 }