NOIP 2021 数列（dp+二进制相关）（读者 rp +++++++）

最新推荐文章于 2024-11-02 09:55:19 发布

L_Aria

最新推荐文章于 2024-11-02 09:55:19 发布

阅读量1.7k

点赞数 33

分类专栏： dp 文章标签：算法动态规划

本文链接：https://blog.csdn.net/L_Aria/article/details/141526378

版权

dp 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

NOIP 2021 数列（dp+二进制相关）（读者 rp +++++++）

序

先剧透下，这道题复杂度 $\Omicron(n^4m)$

快来看看吧

题意

给定 $n, m, k$ 和一个长度为 $\sim m)$ 的正整数数组 $v$

对于一个长度为 $\sim n)$ 的序列 $a$ （ $\le a_i \le m$ ），权值为 $\prod_{i=1}^{n} v_{a_i}$

当序列 $a$ 满足 $popcount(\sum_{i=1}^{n} 2^{a_i}) \le k$ 时¹， $a$ 为一个合法序列。

求所有合法序列的权值和 $\%998244353$ 。

解法

很有意思的一道 dp

先钦定 $a$ 为升序，这样好操作一点，最后再乘上方案数就好。

设 $f_{i,j,k,l}$ 表示选到 $a_i$ （ $a_i$ 已经选完），当前最高位为 $j$ （即最大选了 $j - 1$ ），选了 $k$ 个，状态中有 $l$ 个 $1$ 。

初始化 $f_{0,0,0,0}=1$

首先枚举 $i, j, k, l$ ，然后再枚举一维 $\sim n-i)$ ，意为往后面再放 $p$ 个 $j$

那么 $\sim i+p$ 就全部放好了；最高位也由 $j$ 变成了 $j + 1$ （选得最大的数由 $j - 1$ 变成了 $j$ ）；由于上一位选了 $k$ 个，加上当前选的 $p$ 个，那进位后就变成了 $\lfloor \frac{k+p}{2} \rfloor$ ；同时如果有余数，状态中为 $1$ 的就 $+ 1$ ，所以为 $l+(k+p)\%2$ 。

而因为选了 $p$ 个 $j$ ，所以贡献也要加上 $p$ 个 $j$ 的，所以在输入时预处理，把 $v$ 数组改成二维， $v_{i,j}$ 表示 $i$ 选了 $j$ 个的贡献。

那么状态转移方程即为

f[i+p][j+1][(k+p)>>1][l+(k+p)%2]+=f[i][j][k][l]*v[j][p];

不对不对，还要乘上方案数呢？

上面卡掉，重新来：

f[i+p][j+1][(k+p)>>1][l+(k+p)%2]+=f[i][j][k][l]*v[j][p]*c[i+p][p];

$c$ 就是组合数， $c_{i,j}$ 表示 $C_i^j$ ，可以提前 $\Omicron(n^2)$ 预处理（应该都会，不再赘述，不会的看代码）

那么最后的答案为 $\sum_{i=0}^{n} \sum_{j=0,popcount(i)+j \le K}^{K} f_{n,m+1,i,j}$

代码

#include<cstdio>
#define ll long long int 
using namespace std;
const ll Md=998244353;
const int N=30,M=100;
ll c[N+10][N+10],v[M+10][N+10],f[N+10][M+10][N+10][N+10];
int cnt(int x)
{
	int s=0;
	while(x)
	{
		s+=x%2;
		x>>=1;
	}
	return s;
}
int main()
{
	int n,m,K;
	scanf("%d%d%d",&n,&m,&K);
	for(int i=0;i<=n;i++)
	{
		c[i][0]=1;
		for(int j=1;j<=i;j++)
		{
			c[i][j]=(c[i-1][j]+c[i-1][j-1])%Md;
		}
	}
	for(int i=0;i<=m;i++)
	{
		v[i][0]=1;
		scanf("%lld",&v[i][1]);
		for(int j=2;j<=n;j++)
		{
			v[i][j]=(v[i][j-1]*v[i][1])%Md;
		}
	}
	f[0][0][0][0]=1;
	for(int i=0;i<=n;i++)
	{
		for(int j=0;j<=m;j++)
		{
			for(int k=0;k<=(i>>1);k++)
			{
				for(int l=0;l<=K;l++)
				{
					for(int p=0;p<=n-i;p++)
					{
						f[i+p][j+1][(k+p)>>1][l+(k+p)%2]=(f[i+p][j+1][(k+p)>>1][l+(k+p)%2]+((f[i][j][k][l]*v[j][p])%Md*c[i+p][p])%Md)%Md;
					}
				}
			}
		}
	}
	ll ans=0;
	for(int i=0;i<=n;i++)
	{
		for(int j=0;j<=K;j++)
		{
			if((cnt(i)+j)<=K)
			{
				ans=(ans+f[n][m+1][i][j])%Md;
			}
		}
	}
	printf("%lld",ans);
	return 0;
}