小D的一串数字 (数位DP做法)

 

Problem Description

小D在纸上随便写了两串数字，“23333333”，“0123456789”。第一个串中含有7个重复的相邻的数字，第二个串中不含相邻的重复的数字。小D喜欢第一类串，但是他放低了要求，只要一串数字中含有超过两个重复的相邻的数字，小D就喜欢这串数字。现在他想知道，长度为n的的所有数字串，最多有多少个串会被他喜欢？但是小D是个数学白痴，麻烦你帮他解决这个问题。

Input

输入一个N,(3=<N<=10),代表数字串的长度。

Output

输出一个整数，表示小D最多喜欢多少串。

Sample Input

3

Sample Output

10

Hint

样例中111,222,333,444,555,666,777,888,999，000，这10个串是被小D喜欢的。

#include<bits/stdc++.h>
using namespace std;
int dp[20][11][2][2];
int dfs(int pos,int lian1,int lian2,int lian3)//后三个参数代表有几个位重复的
{
    if(dp[pos][lian1][lian2][lian3])//如果这个数满足条件
    {
        return dp[pos][lian1][lian2][lian3];那么就返回该返回的值
    }
    if(!pos)
    {
        return lian3;//如果到了最后一位，那么返回三个重复的值
    }
    int up=9;//一个位最大数是9
    int sum=0;
    for(int i=0;i<=up;i++)
    {
        if(lian3) sum+=dfs(pos-1,lian1,lian2,lian3);//如果有三个重复的数字，那么直接加上
        else if(i==lian1&&lian2) sum+=dfs(pos-1,lian1,lian2,1);//如果两个重复的数字正好为i那么就让第三位从1开始加
        else if(i==lian1) sum+=dfs(pos-1,i,1,lian3);//如果i等于lian1那么就顺着递归加上两个重复的并且重复的数字为i的情况
        else sum+=dfs(pos-1,i,0,0);//注意函数都是从这里进去的，所以lian2和lian3都是0
    }
    return dp[pos][lian1][lian2][lian3]=sum;

}
int main()
{
    long long n;
    cin>>n;
    cout<<dfs(n,10,0,0)<<endl;//注意刚开始lian1是10
    return 0;
}


/***************************************************
User name: jk180122宋宁
Result: Accepted
Take time: 0ms
Take Memory: 200KB
Submit time: 2019-05-02 21:20:11
****************************************************/