LightOJ1122 Digit Count（DP）

dp[i][j]表示长度i末尾为S[j]的方案数

dp[1][0...m-1]=1

dp[i][j]=∑dp[i-1][k] （|S[k]-S[j]|<=2）

#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
int d[11][11];
int main(){
    int t,n,m,a[11];
    scanf("%d",&t);
    for(int cse=1; cse<=t; ++cse){
        scanf("%d%d",&n,&m);
        for(int i=0; i<n; ++i) scanf("%d",a+i);
        memset(d,0,sizeof(d));
        for(int i=0; i<n; ++i) d[1][i]=1;
        for(int i=2; i<=m; ++i){
            for(int j=0; j<n; ++j){
                for(int k=0; k<n; ++k){
                    if(abs(a[j]-a[k])<=2) d[i][j]+=d[i-1][k];
                }
            }
        }
        int res=0;
        for(int i=0; i<n; ++i) res+=d[m][i];
        printf("Case %d: %d
",cse,res);
    }
    return 0;
}