Alice's Piano_回溯

 题目描述：

    弹钢琴的题。给两个0,1序列。目标是从任一一列开始走，走的过程中可以随时选择走上一列或者走下一列，保证走的过程中，按到的黑白键的数目差值不超过k。从头走到尾，两列都走到尾部时结束。

 

解题思路：

   显然回溯。然后问题是，，不负众望的tle出现了。需要优化。能够优化的地方，就是记录每次失败所处状态，使得下一次回溯回退到同样状态时不需要再进行探索，或者在某一次探索到该状态时就知道不会成功，直接回溯回去。

 

代码：

#include <stdio.h>
#include <stdlib.h>
#define N_MAX 1001

int piano[3][N_MAX],p[2*N_MAX];
int N,K;
int flag[N_MAX][N_MAX];

int press(int num, int r1, int r2, int k)
{
   int f;
   if(flag[r1][r2] == 0)
         return 0;
   if(num == 2*N)
      return 1;
   else
   {
      if((r1+1 <= N) &&((piano[1][r1+1] == 0 && k<K)||(piano[1][r1+1] == 1 && k>-K)))
      {

         p[num+1] = 1;
         if( 0 == piano[1][r1+1] )
            f = press(num+1,r1+1,r2,k+1);
         else
            f = press(num+1,r1+1,r2,k-1);
         if(f == 1)
            return 1;

      }
      if((r2+1 <= N) && ((piano[2][r2+1] == 0 && k<K)||(piano[2][r2+1] == 1 && k>-K)))
      {
       
         p[num+1] = 2;
         if( 0 == piano[2][r2+1] )
            f = press(num+1,r1,r2+1,k+1);
         else
            f = press(num+1,r1,r2+1,k-1);
         if(f == 1)
            return 1;
      }
   }
   flag[r1][r2] = 0;
   return 0;
}

main()
{
   int i,f,j;
   char str[N_MAX];
  
   while(scanf("%d %d",&N,&K)!=EOF)
   {
       scanf("%s",str);
       for(i=1;i<=N;i++)
          piano[1][i] = str[i-1] -'0';
       scanf("%s",str);
       for(i=1;i<=N;i++)
         piano[2][i] = str[i-1] -'0';
        
       for(i=0;i<=N;i++)
          for(j=0;j<=N;j++)
             flag[i][j] = 1;
   
       f = press(0,0,0,0);     
       if(1 == f)
       {
          for(i=1;i<=2*N;i++)
             printf("%d",p[i]);
          printf("\n");
       }
       else
          printf("Poor Alice\n");
   }

   //system("pause");
   return 0;  
}