Codeforces Round 871 (Div. 4)

Problem - G - Codeforces

思路：

1. 设i位于第len行的第p个位置
   1. 发现p=1，dp[i]=i*i+dp[i-len+1]
   2. p=len,dp[i]=i*i+dp[i-len]
   3. 其余情况，dp[i]可有他上面的i-len与i-len+1推出来，但是i-len与i-len+1两人的上面的i-len-(len-1)+1是重复了的，所以算了两遍，减去一次即可

#include <bits/stdc++.h>
using namespace std;
#define ll               long long
#define endl             "\n"
#define int              long long
#define endll            endl<<endl
typedef unsigned long long ull;
typedef pair<long long, long long> pll;
//---------------------------------------------------------------------------------------------------------------------//
//---------------------------------------------------------------------------------------------------------------------//
const int INF = 0x3f3f3f3f;         //int型的INF
const ll llINF = 0x3f3f3f3f3f3f3f3f;//ll型的llINF
const int N = 1e6+ 10;

int dp[N];
void mysolve()
{
	int n;
	cin>>n;
	cout<<dp[n]<<endl;
}

int32_t main()
{
	std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	ll t=1;
	cin >> t;
	dp[1]=1;
	int len=2,p=1;
	for(int i=2; i<=1e6; ++i)//预处理
		{
			if(p==1)dp[i]=i*i+dp[i-len+1];
			else if(p==len)dp[i]=i*i+dp[i-len];
			else
				{
					dp[i]=i*i+dp[i-len+1]+dp[i-len]-dp[i-len-(len-1)+1];//要减去上方重复的
				}
			p++;
			if(p>len)len++,p=1;
		}
	while (t--)
		{
			mysolve();
		}
	system("pause");
	return 0;
}

Problem - H - Codeforces

思路：

1. 显然子序列问题用dp推导即可。用dp[i][j]表示处理到第i个数，状态为j时的方案数。
2. 因为要求不可以有空串，所以k==6时，因为我们初始的dp[0][(1<<6)-1]=1，等于多算了一次（即这个空串本来是无意义的，但是他本身1的个数就是6，所以算了一次他（如果n个a全部不取，就会算他一次），所以减去一次

#include <bits/stdc++.h>
using namespace std;
#define ll               long long
#define endl             "\n"
#define int              long long
const int N = 2e5 + 10;
const int mod=1e9+7;
int cnt[100];
int a[N];
void mysolve()
{
	int n,k;
	cin>>n>>k;
	for(int i=1; i<=n; ++i)cin>>a[i];
	vector<vector<int>>dp(n+1,vector<int>(100));
	dp[0][(1<<6)-1]=1;
	for(int i=1; i<=n; ++i)for(int j=0; j<64; ++j)if(dp[i-1][j]>0)
				{
					dp[i][j]=(dp[i][j]+dp[i-1][j])%mod;
					dp[i][j&a[i]]=(dp[i][j&a[i]]+dp[i-1][j])%mod;
				}
	int ans=0;
	for(int i=0; i<64; ++i)if(cnt[i]==k)
			ans=(ans+dp[n][i])%mod;
	if(k==6)ans=(ans-1+mod)%mod;
	cout<<ans<<endl;
}

int32_t main()
{
	std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	ll t=1;
	for(int i=0; i<(1<<6); ++i)for(int j=0; j<6; ++j)if(i&(1<<j))cnt[i]++;
	cin >> t;
	while (t--)
		{
			mysolve();
		}
	system("pause");
	return 0;
}