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;
}
京公网安备 11010502036488号