AcWing 3195. 有趣的数（枚举 + 组合计数）

题目链接

我们把一个数称为有趣的，当且仅当：

1. 它的数字只包含 0,1,2,3，且这四个数字都出现过至少一次。
2. 所有的 0 都出现在所有的 1 之前，而所有的 2 都出现在所有的 3 之前。
3. 最高位数字不为 0。

因此，符合我们定义的最小的有趣的数是 2013。

除此以外，4 位的有趣的数还有两个：2031 和 2301。

请计算恰好有 n 位的有趣的数的个数。

由于答案可能非常大，只需要输出答案除以 109+7 的余数。

输入格式

输入只有一行，包括恰好一个正整数 n。

输出格式

输出只有一行，包括恰好 n 位的整数中有趣的数的个数除以 109+7 的余数。

数据范围

4≤n≤1000

输入样例：

4

输出样例：

3

答案：

#include <iostream>
#include<bits/stdc++.h>
#define ll long long
#define ull unsigned long long
#define INF 0x3f3f3f3f
#define inf 0x3f3f3f3f3f3f3f3f
#define rep(i,a,b) for(auto i=a;i<=b;++i)
#define bep(i,a,b) for(auto i=a;i>=b;--i)
#define lowbit(x) x&(-x)
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define PI acos(-1)
#define pb push_back
#define eps 1e-6
const int mod = 1e9 + 7;
const int MOD = 1e4+7;
const int N = 4e5 + 10;
const int M = 1111;
int dx[]={-1, 0, 1, 0};
int dy[]={0, 1, 0, -1};
int dxy[][2]={{0,1},{1,0},{1,1},{-1,1}};
using namespace std;

ll dp[M][7];

void solve(){
    ll n;
    cin>>n;
    rep(i,0,5) dp[0][i]=0;
    rep(i,1,n){
        ll j=i-1;
        dp[i][0]=1;
        dp[i][1]=(dp[j][0]+dp[j][1]*2)%mod;
        dp[i][2]=(dp[j][0]+dp[j][2])%mod;
        dp[i][3]=(dp[j][1]+dp[j][3]*2)%mod;
        dp[i][4]=(dp[j][1]+dp[j][2]+dp[j][4]*2)%mod;
        dp[i][5]=(dp[j][3]+dp[j][4]+dp[j][5]*2)%mod;
    }
    cout<<dp[n][5]<<endl;
}

int main() {
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    solve();
	return 0;
}