HDU 5898 odd-even number

题目：odd-even number

链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=5898

题意：给一个条件，问l 到r 之间有多少满足条件的数，条件是：连续的奇数长度为偶数，连续的偶数长度为奇数，比如124683，连续的奇数（1、3）长度都是1（奇数），连续的偶数（2468）长度为4（偶数），所以不满足条件。

思路：

　　很明显的数位dp，可以用dfs做，传下三个参数（pre、p1、p2、ceng），pre表示前一位的数是奇数还是偶数，p1表示前面连续的奇数个数，p2表示前面连续的偶数的个数（p1、p2至少有一个为0），ceng表示还需dfs几位。如果ceng是0，那就判断，pre是偶数p2为奇数、pre是奇数p1是偶数返回1，否则返回0。如果ceng不为0，则分情况递归，比如pre为1且p1为奇数，则这一位不能选择偶数......

　　然后就是根据具体的数调用dfs求数量。

　　注意：区间超过longlong，要用字符串存。

AC代码：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>
#include<set>
#include<map>
#include<list>
#include<stack>
#include<queue>
#include<vector>
#include<string>
#include<iostream>
#include<algorithm>
using namespace std;
#define lson rt<<1
#define rson rt<<1|1
#define N 20020
#define M 100010
#define Mod 1000000007
#define LL long long
#define INF 0x7fffffff
#define FOR(i,f_start,f_end) for(int i=f_start;i<=f_end;i++)
#define For(i,f_start,f_end) for(int i=f_start;i<f_end;i++)
#define REP(i,f_end,f_start) for(int i=f_end;i>=f_start;i--)
#define Rep(i,f_end,f_start) for(int i=f_end;i>f_start;i--)
#define MT(x,i) memset(x,i,sizeof(x))
#define gcd(x,y) __gcd(x,y)
const double PI = acos(-1);

LL dfs(int pre,int p1,int p2,int ceng)
{
  if(ceng == 0)
  {
    if(p1==0 && p2==0) return 1;
    else if(pre==0 && p2%2==1) return 1;
    else if(pre==0) return 0;
    else if(pre==1 && p1%2==0) return 1;
    else return 0;
  }
  LL ret;
  if(p1==0 && p2==0){
    ret = dfs(0,0,0,ceng-1);
    ret += dfs(0,0,p2+1,ceng-1)*4;
    ret += dfs(1,p1+1,0,ceng-1)*5;

  }
  else if(pre==0 && p2%2==0){
    ret=dfs(0,0,p2+1,ceng-1);
    ret = ret * 5;
  }
  else if(pre==0 && p2%2==1){
    ret=dfs(0,0,p2+1,ceng-1)*5;
    ret+=dfs(1,p1+1,0,ceng-1)*5;
  }
  else if(pre==1 && p1%2==1){
    ret=dfs(1,p1+1,0,ceng-1)*5;
  }
  else if(pre==1 && p1%2==0){
    ret=dfs(1,p1+1,0,ceng-1)*5;
    ret+=dfs(0,0,p2+1,ceng-1)*5;
  }
  return ret;
}

bool ok(char *x)
{
  if(x[0]=='0' && x[1]==0) return true;
  int pre = 0, p1 = 0, p2 = 0;
  int len=strlen(x)-1;
  while(len>=0)
  {
    int tmp = (x[len]-'0')%2;
    if(p1==0 && p2==0);
    else if(tmp == 0 && pre==1 && p1%2==1) return false;
    else if(tmp == 1 && pre==0 && p2%2==0) return false;
    if(tmp == 0 ) p2++,p1=0;
    else p1++,p2=0;
    pre=tmp;
    len--;
  }
  if(pre==0 && p2%2==0) return false;
  if(pre==1 && p1%2==1) return false;
  return true;
}

bool ok(int x)
{
  if(x==0) return true;
  int pre = 0, p1 = 0, p2 = 0;
  while(x)
  {
    int tmp = x%10%2;
    if(p1==0 && p2==0);
    else if(tmp == 0 && pre==1 && p1%2==1) return false;
    else if(tmp == 1 && pre==0 && p2%2==0) return false;
    if(tmp == 0 ) p2++,p1=0;
    else p1++,p2=0;
    pre=tmp;
    x/=10;
  }
  if(pre==0 && p2%2==0) return false;
  if(pre==1 && p1%2==1) return false;
  return true;
}

LL solve(char *x)
{
  LL ret = 0;
  if(ok(x)){
    ret++;
  }
  int bt[100],bo=0;
  while(x[bo])
  {
    bt[bo]=x[bo]-'0';
    bo++;
  }
  int tmp = 0;
  while(tmp<=(bo-1)/2){
    int tt = bt[tmp];
    bt[tmp]=bt[bo-tmp-1];
    bt[bo-tmp-1]=tt;
    tmp++;
  }
  int pre = 0, p1 = 0, p2 = 0;
  for(int i=bo-1;i>=0;i--)
  {
    if(bt[i]>0)
    {
      if(p1 == 0 && p2==0)
        ret += dfs(pre,0,0,i);
      else
      {
        if(pre == 1 && p1 % 2 == 1 );
        else ret += dfs(0,0,p2+1,i);
      }
    }
    for(int j=1;j<bt[i];j++)
    {
      if(j%2==1){
        if((p1!=0 || p2!=0) && pre == 0 && p2 % 2 == 0) continue;
        else ret += dfs(1,p1+1,0,i);
      }
      else
      {
        if(pre == 1 && p1 % 2 == 1) continue;
        ret += dfs(0,0,p2+1,i);
      }
    }
    if(p1==0 && p2==0);
    else if(bt[i]%2==0 && pre==1 && p1%2==1) break;
    else if(bt[i]%2==1 && pre==0 && p2%2==0) break;
    pre = bt[i]%2;
    if(pre==0) p2++,p1=0;
    else p1++,p2=0;
  }
  return ret;
}

int main()
{
  int t;
  char l[100],r[100];
  int cas=1;
  scanf("%d",&t);
  while(t--){
    scanf("%s%s",l,r);
    printf("Case #%d: ",cas++);
    printf("%I64d
",solve(r)-solve(l)+(ok(l)?1:0));
  }
  return 0;
}