cf1158F

cf1158F - Density of subarrays

自己想出来的cf div1F:)

题解留坑待补(其实也就是个套路的找出某种定义对于的一个结构的一种思路吧。。。不过以前基本都肝不出来呢。。。算是有点进步吧。。。虽然我已经弱到被认识的大部分爆踩了qwq)

#include<bits/stdc++.h>
using namespace std;
const int N=3005,P=998244353;
int n,c,a[N];
namespace sol1
{
	int f[N][N];
	inline void sol()
	{
		int nn=(1<<c)-1;f[0][0]=1;
		for(int i=1;i<=n;++i)for(int j=n/c;~j;--j)for(int k=nn-1;~k;--k)if(f[j][k])
		{
			if((k|1<<a[i])==nn)f[j+1][0]=(f[j+1][0]+f[j][k])%P;
			else f[j][k|1<<a[i]]=(f[j][k|1<<a[i]]+f[j][k])%P;
		}
		for(int i=0;i<=n;++i)
		{
			int ans=i?0:P-1;
			for(int j=0;j<nn;++j)ans=(ans+f[i][j])%P;
			printf("%d ",ans);
		}
	}
}
namespace sol2
{
	int w[N],iv[N],s[N][N],f[N][N],ans[N];
	inline int pw(int a,int s){int r=1;for(;s;s>>=1,a=1ll*a*a%P)if(s&1)r=1ll*r*a%P;return r;}
	void sol()
	{
		for(int i=1;i<=n;++i){for(int j=0;j<c;++j)s[i][j]=s[i-1][j];s[i][a[i]]++;}
		for(int i=1;i<=n;++i)w[i]=pw(2,i)-1,iv[i]=pw(w[i],P-2);
		f[0][0]=1;
		for(int i=0;i<=n;++i)
		{
			int cc=0,res=1,tt=w[n-i]+1;
			for(int j=i+1;j<=n;++j)
			{
				int d=s[j][a[j]]-s[i][a[j]];
				if(d>1)res=1ll*res*w[d]%P*iv[d-1]%P;else cc++;
				if(cc==c)
				{
					res=1ll*res*iv[d]%P;tt=(tt+1ll*(P-res)*(w[n-j]+1))%P;
					for(int k=i/c;~k;--k)f[j][k+1]=(f[j][k+1]+1ll*f[i][k]*res)%P;
					res=1ll*res*w[d]%P;
				}
			}
			for(int j=i/c;~j;--j)ans[j]=(ans[j]+1ll*f[i][j]*tt)%P;
		}
		ans[0]=(ans[0]+P-1)%P;
		for(int i=0;i<=n;++i)printf("%d ",ans[i]);
	}
}
int main()
{
	scanf("%d%d",&n,&c);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]),a[i]--;
	c<=11?sol1::sol():sol2::sol();return 0;
}