洛谷3600 随机数生成器

Problem

洛谷

整数期望公式

$E(x)={\sum }_i^{\infty }P(x\ge i)$
这个和我们常用的$E(x)={\sum }_i^{\infty }P(i)\ast i$其实是等价的，因为$P(x\ge i)={\sum }_{j=i}^{\infty }P(j)$，这就相当于把原来的$P(i)\ast i$拆成i份，加到前面的P之中。

Solution

$E(ans)={\sum }_{i=1}^{x}1-P(ans<i)$。
显然对于一个询问区间包含另一个询问区间的情况下，前者是可以直接舍弃的，因为前者的$\min \le$后者的$\min$。
为了避免重叠区间的相互影响，我们对序列中的元素进行考虑。那么可以预处理出覆盖到点i的询问，显然它应该是一段连续的区间，记为$[pre[i],nxt[i]]$。而第i个点有$\frac{i-1}{x}$的概率会小于i，对覆盖到它的区间做贡献，使得这些区间满足条件。

不妨设$f[i]$表示位置i做贡献且1~pre[i]-1的询问全都已经满足条件的概率。枚举上一个做贡献的点，则有方程

$$f[i]=p\ast ([pre[i]==1]\ast (1-p{)}^{i-1}+\sum _{nxt[j]\ge pre[i]-1}f[j]\ast (1-p{)}^{i-j-1})$$

最后的答案就是${\sum }_{nxt[i]=q}f[i]\ast (1-p{)}^{n-i}$

由于区间的$l$和$r$都分别是递增的，$two-pointer$一下即可做到$O(n)$的DP了。
时间复杂度$O(nx)$

---

好像设$f[i]$表示右端点小于等于$i$的区间均满足条件也可以做，但是我不会转移= =||

Code

#include <algorithm>
#include <cstdio>
using namespace std;
typedef long long ll;
const int maxn=2010,mod=666623333;
struct data{
	int l,r;
	bool operator < (const data &x)const
	{
		if(l==x.l) return r>x.r;
		return l<x.l;
	}
}a[maxn];
struct Stack{
	int tp,a[maxn];
	Stack(){tp=0;}
	void push(int val){a[++tp]=val;}
	int top(){return a[tp];}
	void pop(){tp--;}
}stk;
int n,x,q,p,pc,ans,pre[maxn],nxt[maxn],inv[maxn],f[maxn];
int power(int x,int y)
{
	int res=1;
	while(y)
	{
		if(y&1) res=(ll)res*x%mod;
		x=(ll)x*x%mod;y>>=1;
	}
	return res;
}
void input()
{
	int head=1,tail=0;
	scanf("%d%d%d",&n,&x,&q);
	for(int i=1;i<=q;i++) scanf("%d%d",&a[i].l,&a[i].r);
	sort(a+1,a+q+1);
	for(int i=1;i<=q;i++)
	{
		while(a[i].r<=a[stk.top()].r) stk.pop();
		stk.push(i);
	}
	q=stk.tp;
	for(int i=1;i<=q;i++) a[i]=a[stk.a[i]];
	for(int i=1;i<=n;i++)
	{
		while(tail<q&&a[tail+1].l<=i) tail++;
		while(head<=tail&&a[head].r<i) head++;
		pre[i]=head;nxt[i]=tail;
	}
}
int main()
{
	#ifndef ONLINE_JUDGE
	freopen("in.txt","r",stdin);
	#endif
	input();
	for(int i=1;i<=x;i++)
	{
		int j=0,s=1,t=1;
		inv[0]=1;f[0]=1;
		p=((ll)(i-1+mod)%mod*power(x,mod-2))%mod;
		pc=(1-p+mod)%mod;inv[1]=power(pc,mod-2);
		for(int k=2;k<=n;k++) inv[k]=(ll)inv[k-1]*inv[1]%mod;
		for(int k=1;k<=n;k++)
		{
			while(j<k&&nxt[j]<pre[k]-1)
			  s=(s-((ll)f[j]*inv[j]%mod)+mod)%mod,j++;
			f[k]=(ll)s*t%mod*p%mod;t=(ll)t*pc%mod;
			s=(s+(ll)f[k]*inv[k])%mod;
		}
		s=0;t=1;
		for(int k=n;k>=1&&nxt[k]==q;k--)
		  s=(s+(ll)f[k]*t%mod)%mod,t=(ll)t*pc%mod;
		ans+=(1-s+mod)%mod;
		if(ans>=mod) ans-=mod;
	}
	printf("%d\n",ans);
	return 0;
}