CodeForces 44 H.Phone Number（dp）

Description

给出一个长度为$n$的数字串$a_1...a_n$，首先随机确定一个数字$b_1(0\le b_i\le 9)$，之后令$b_i=\lfloor \frac{b_{i-1}+a_{i}}{2}\rfloor$或$\lceil \frac{b_{i-1}+a_{i}}{2}\rceil$，$2\le i\le n$，以此确定数字串$b$，问不同的$b$串数量（不包括$a$）

Input

一个长度不超过$50$的数字串$a$

Output

输出不同的$b$串数量

Sample Input

12345

Sample Output

48

Solution

以$dp[i][j]$表示使得$b_i=j$的$b_1...b_i$个数，则$dp[1][0...9]=1$，令$k=\lfloor \frac{j+a_{i+1}}{2}\rfloor,l=\lceil\frac{j+a_{i+1}}{2}\rceil$，根据第$b_{i+1}$的选取有转移：

$dp[i+1][k]+=dp[i][j],dp[i+1][l]+=dp[i][j](l\neq k)$

$\sum\limits_{i=0}^{9}dp[n][i]$即为答案，注意如果$a$序列满足$a_i=\lfloor\frac{a_{i-1}+a_i}{2}\rfloor$或$a_i=\lceil\frac{a_{i-1}+a_i}{2}\rceil$则答案需要减一

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=100001;
int n,a[55];
ll dp[55][10];
char s[55];
int main()
{
    scanf("%s",s+1);
    n=strlen(s+1);
    for(int i=1;i<=n;i++)a[i]=s[i]-'0';
    for(int i=0;i<10;i++)dp[1][i]=1;
    for(int i=1;i<n;i++)
        for(int j=0;j<10;j++)
        {
            int k=(j+a[i+1])/2,l=(j+a[i+1]+1)/2;
            dp[i+1][k]+=dp[i][j];
            if(k!=l)dp[i+1][l]+=dp[i][j];
        }
    ll ans=0;
    for(int i=0;i<10;i++)ans+=dp[n][i];
    int flag=1;
    for(int i=2;i<=n;i++)
    {
        if((a[i-1]+a[i])/2==a[i]||(a[i-1]+a[i]+1)/2==a[i])continue;
        else    
        {
            flag=0;
            break;
        }
    }
    printf("%I64d\n",ans-flag);
    return 0;
}