牛客练习赛C (线性dp)

https://ac.nowcoder.com/acm/contest/43058/C

一般这种求子序列方案数的都是一些线性dp, 我们可以定义f[i][j]为最后一位数字为i，且当前长度为j的子序列的个数。

显然有两种转移方式，一种是j=2或4，另一种是其他情况.

我一开始写的是：

for(int j=7;j>=2;j--){
			if(j==2||j==4){
				for(int k=t+1;k<=9;k++) f[k][j]=(f[k][j]+f[t][j-1])%mod;
			}
			else {
				for(int k=0;k<t;k++) f[k][j]=(f[k][j]+f[t][j-1])%mod;
			}
		}

然后就wa了，开始不知道怎么回事，然后发现是转移的方式有问题，正确的代码是这样的：

 for(int j=7;j>=2;j--){
            if(j==2||j==4){
                for(int k=0;k<t;k++) f[t][j]=(f[t][j]+f[k][j-1])%mod;
            }
            else {
                for(int k=t+1;k<=9;k++) f[t][j]=(f[t][j]+f[k][j-1])%mod;
            }
        }

很明显第一个代码错的原因是没有按照状态的定义来转移。

完整代码如下：

#include<bits/stdc++.h>

using namespace std;
#define int long long

const int mod=1e9+7;
int f[20][20];
signed main()
{
	string s;
	cin>>s;
	int n=s.size();
	s=" "+s;
	for(int i=1;i<=n;i++){
		int t=s[i]-'0';
		f[t][1]++;
		for(int j=7;j>=2;j--){
			if(j==2||j==4){
				for(int k=0;k<t;k++) f[t][j]=(f[t][j]+f[k][j-1])%mod;
			}
			else {
				for(int k=t+1;k<=9;k++) f[t][j]=(f[t][j]+f[k][j-1])%mod;
			}
		}
	}
	int ans=0;
	for(int i=0;i<=9;i++){
		ans=(ans+f[i][7])%mod;
	}
	cout<<ans<<endl;
	return 0;
} 

我好菜！！！