酸菜鱼的 DP动态规划 刷题记录

BZOJ1026: [SCOI2009]windy数

• 数位dp。很多小细节。。。
• 代码：

  #include <bits/stdc++.h>
   
  using namespace std;
  typedef long long ll;
  char a[30],b[30];
  ll d[30][20]={0};
  int la,lb;
   
  void pre(){
      for (int j=0; j<=9; j++) d[1][j]=1;
      for (int i=2; i<=lb; i++)
          for (int j=0; j<=9; j++)
              for (int k=0; k<=9; k++)
                  if(abs(j-k)>=2) d[i][j]+=d[i-1][k];
  }
   
  ll solve(char* s,int l){
      ll ans=0;
      //先算位数小的
      for (int i=1; i<l; i++) for (int j=1; j<=9; j++) ans+=d[i][j];
      //算位数相等的
      for (int i=1; i<s[1]-'0'; i++) ans+=d[l][i];
      for (int i=2; i<=l; i++) {  //前面都是填的相等的数
          for (int j=0; j<(s[i]-'0'); j++)   //第i位填j,第i+1位填k
              if( abs( j - (s[i-1]-'0') ) >= 2 ) ans+=d[l-i+1][j];
          if(abs(s[i]-s[i-1])<2) break;  //不要忘记这个跳出，不合法了都
      }
      return ans;
  }
   
  bool judge(){  //判断大的那个数是不是
      bool flag=true;
      for (int i=2; i<=lb; i++) if(abs(b[i]-b[i-1])<=2) { flag=false; break; }
      return flag;
  }
   
  int main(){
      scanf("%s%s",a+1,b+1);
      la=strlen(a+1);
      lb=strlen(b+1);
      pre();
      ll ansa=solve(a,la);
      ll ansb=solve(b,lb);
      if(judge()) ansb++;
      cout<<ansb-ansa<<endl;
      return 0;
  }