luogu P3600 随机数生成器【dp】

把期望改成方案数最后除一下,设h[i]为最大值恰好是i的方案数,那么要求的就是Σh[i]*i
首先包含其他区间的区间是没有意义的,用单调栈去掉
然后恰好不好求,就改成h[i]表示最大值最大是i的方案数,求Σ(h[i]-h[i-1])*i即可
然后考虑h怎么求,\( h[i]=\sum_{j=1}{n}ij*(m-1)^{n-j}*选j个点使得每个区间都有一个选中点 \)
设选j个点使得每个区间都有一个选中点为g[j],设f[i][j]为前i个必选i点一共选j个点的方案数,那么\( g[j]=\sum_{fr[i]==q}f[i][j] \)
然后考虑f怎么求,设fl[i]fr[i]分别是包含i的最左/右区间(如果没被包含就是左边第一个和右边第一个),\( f[i][j]=\sum_{fr[k]+1>=fl[i]} f[k][j-1] \),前缀和优化即可

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=2005,mod=666623333;
int n,m,q,fl[N],fr[N],f[N][N],s[N][N],g[N],h[N],top,ans;
struct qwe
{
	int l,r;
}a[N],b[N];
bool cmp(const qwe &a,const qwe &b)
{
	return a.l<b.l||(a.l==b.l&&a.r<b.r);
}
int read()
{
	int r=0,f=1;
	char p=getchar();
	while(p>'9'||p<'0')
	{
		if(p=='-')
			f=-1;
		p=getchar();
	}
	while(p>='0'&&p<='9')
	{
		r=r*10+p-48;
		p=getchar();
	}
	return r*f;
}
int ksm(int a,int b)
{
	int r=1;
	while(b)
	{
		if(b&1)
			r=1ll*r*a%mod;
		a=1ll*a*a%mod;
		b>>=1;
	}
	return r;
}
int main()
{
	n=read(),m=read(),q=read();
	for(int i=1;i<=q;i++)
		a[i].l=read(),a[i].r=read();
	sort(a+1,a+1+q,cmp);
	for(int i=1;i<=q;i++)
		if(i==1||a[i].l!=a[i-1].l)
		{
			while(top&&b[top].r>=a[i].r)
				top--;
			b[++top]=a[i];//cerr<<i<<endl;
		}
	q=top;
	// for(int i=1;i<=q;i++)
		// cerr<<b[i].l<<"   "<<b[i].r<<endl;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=q;j++)
			if(b[j].r>=i)
			{
				fl[i]=j;
				break;
			}
		for(int j=q;j>=1;j--)
			if(b[j].l<=i)
			{
				fr[i]=j;
				break;
			}
		if(!fl[i])
			fl[i]=fr[i]+1;
		// cerr<<fl[i]<<" "<<fr[i]<<endl;
	}
	f[0][0]=s[0][0]=1;
	top=0;
	for(int i=1;i<=n;i++)
	{
		s[i][0]=s[i-1][0];
		for(int j=1;j<=i;j++)
		{
			while(top<i-1&&fr[top]+1<fl[i])
				top++;
			f[i][j]=(s[i-1][j-1]-(top?s[top-1][j-1]:0)+mod)%mod;
			s[i][j]=(s[i-1][j]+f[i][j])%mod;
		}
	}
	for(int j=1;j<=n;j++)
		for(int i=1;i<=n;i++)
			if(fr[i]==q)
				g[j]=(g[j]+f[i][j])%mod;
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++)
			h[i]=(h[i]+1ll*g[j]*ksm(i,j)%mod*ksm(m-i,n-j)%mod)%mod;
		// cerr<<h[i]<<endl;
		ans=(ans+1ll*(h[i]-h[i-1]+mod)*i%mod)%mod;
	}
	printf("%lld\n",1ll*ans*ksm(ksm(m,n),mod-2)%mod);
	return 0;
}
posted @ 2019-05-08 16:29  lokiii  阅读(164)  评论(0编辑  收藏  举报