codeforces 1288C Two Arrays

题目

思路

1. Dp思想，将两个序列合并变成$a_1,a_2,...a_m,{\mathbf{b}}_{\mathbf{m}},{\mathbf{b}}_{\mathbf{m}-1},...\mathbf{b}1$那么这个序列只要单调不减就可以了。
2. 所以，问题转化成了一个序列中元素的取值在$[1,n]$范围内，满足单调不减的序列有多少种
3. 我们可以设$dp[i][j]$表示序列在第$i$个位置，取值小于等于$j$的方案数
4. 我们可以列出方程:

dp[i][j]=dp[i-1][j]+dp[i-1][j-1]+...+dp[i-1][1];
dp[i][j-1]=dp[i-1][j-1]+dp[i-1][j-2]+...+dp[i-1][1];
两个方程相减合并可以得到优化后的方程
dp[i][j]=dp[i-1][j]+dp[i][j-1] j从1开始计数

代码

#include<bits/stdc++.h>

using namespace std;
#define ll long long
const ll mod = 1e9+7;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int n, m;
    cin >> n >> m;
    ll ans = 0;
    vector<vector<int>> dp(2*m+1,vector<int>(n+2,0));
    for(int i=1;i<=2*m;i++){
        for(int j=1;j<=n;j++){
            if(i==1) dp[i][j]=1;
            else{
                dp[i][j]=(dp[i-1][j]+dp[i][j-1])%mod;
            }
            if(i==2*m) ans=(ans+dp[i][j])%mod;
        }
    }
    cout << ans % mod << endl;
    return 0;
}