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");
   return0;  
}