C题
写的有点丑
复杂度
每次考虑前i个数中
1234号人选 4 张不同的牌有几种做法记为dp1234[i]
123号人选 3 张不同的牌有几种做法dp123[i]
之后dp12[i]。。。。dp1[1]的含义以此类推
假如第1个人有第 i 张牌
那么需要加上第一个人拿第 i 张牌,其余人拿前 i-1 张牌的个数,即
同时也要更新下面几个,后面的以此类推,所以写了一堆。。。
dp1[i] += 1;
dp12[i] += dp2[i - 1];
dp13[i] += dp3[i - 1];
dp14[i] += dp4[i - 1];
dp123[i] += dp23[i - 1];
dp124[i] += dp24[i - 1];
dp134[i] += dp34[i - 1];
假如第2个人有第i张牌,那么最终结果需要加上134号人前i-1个数中选3个不同的个数
假如第3个人有第i张牌,那么最终结果需要加上124号人前i-1个数中选3个不同的个数
假如第4个人有第i张牌,那么最终结果需要加上123号人前i-1个数中选3个不同的个数
#include<iostream> #include<queue> #include<vector> #include<map> #include<unordered_map> #include<stack> #include<set> #include<algorithm> #include<cmath> #include<cstring> #define INF 0x3f3f3f3f using namespace std; typedef long long ll; typedef pair<int,int> PII; typedef pair<ll,ll> PLL; const int N = 5e3 + 10; char c[10][5005]; int n; ll dp1234[N]; ll dp123[N]; ll dp124[N]; ll dp134[N]; ll dp234[N]; ll dp12[N]; ll dp13[N]; ll dp14[N]; ll dp23[N]; ll dp24[N]; ll dp34[N]; ll dp1[N]; ll dp2[N]; ll dp3[N]; ll dp4[N]; void init(int i) { dp1234[i] = dp1234[i-1]; dp123[i] = dp123[i-1]; dp124[i] = dp124[i-1]; dp134[i] =dp134[i-1]; dp234[i] = dp234[i-1]; dp12[i]=dp12[i-1]; dp13[i]=dp13[i-1]; dp14[i]=dp14[i-1]; dp23[i]=dp23[i-1]; dp24[i]=dp24[i-1]; dp34[i]=dp34[i-1]; dp1[i]=dp1[i-1]; dp2[i]=dp2[i-1]; dp3[i]=dp3[i-1]; dp4[i]=dp4[i-1]; } int main() { cin >> n; for (int i = 1; i <= 4;i++) scanf("%s", c[i] + 1); for (int i = 1; i <= n;i++) { init(i); if(c[1][i]=='1') { dp1[i] += 1; dp12[i] += dp2[i - 1]; dp13[i] += dp3[i - 1]; dp14[i] += dp4[i - 1]; dp123[i] += dp23[i - 1]; dp124[i] += dp24[i - 1]; dp134[i] += dp34[i - 1]; dp1234[i] += dp234[i - 1]; } if(c[2][i]=='1') { dp2[i] += 1; dp12[i] += dp1[i - 1]; dp23[i] += dp3[i - 1]; dp24[i] += dp4[i - 1]; dp234[i] += dp34[i - 1]; dp123[i] += dp13[i - 1]; dp124[i] += dp14[i - 1]; dp1234[i] += dp134[i - 1]; } if(c[3][i]=='1') { dp3[i] += 1; dp13[i] += dp1[i - 1]; dp23[i] += dp2[i - 1]; dp34[i] += dp4[i - 1]; dp234[i] += dp24[i - 1]; dp123[i] += dp12[i - 1]; dp134[i] += dp14[i - 1]; dp1234[i] += dp124[i - 1]; } if(c[4][i]=='1') { dp4[i] += 1; dp14[i] += dp1[i - 1]; dp34[i] += dp3[i - 1]; dp24[i] += dp2[i - 1]; dp234[i] += dp23[i - 1]; dp124[i] += dp12[i - 1]; dp134[i] += dp13[i - 1]; dp1234[i] += dp123[i - 1]; } } ll res = dp1234[n]; cout << res <<endl; return 0; }