[USACO 2020 February Platinum]Help Yourself

[USACO 2020 February Platinum]Help Yourself

真的很套路。。。

考虑将区间((L_i,R_i))按照左端点排序,依次考虑每个区间的贡献

(dp_i)表示当前所有选择的右端点中最大的为(i)时的方案数

加入区间((L,R))

1.所有(i<L)的部分一定会断开成两个区间,转移时个数+1

2.当(ige L)时,(dp_i)(dp_{maxlbrace R,i brace})转移,分两类讨论即可

不考虑个数的问题,直接转移是(O(n^2))的,但是可以用线段树优化到(nlog n)

(比较麻烦,需要实现区间查询,单点修改,区间乘法)

考虑个数(k)次幂,一种暴力的办法是存下(dp_{i,j}),但是转移会变成(O(n^2log n))

对于当前个数为(c)的情况,如果新增一个联通块,即变为(c+1),答案由(c^k) 变为((c+1)^k)

考虑直接用二项式定理展开这个式子,需要记录(c^i(iin [0,k]))的所有答案,再(O(k^2))完成+1操作

结合线段树,维护(nk)个值,复杂度为(O(n(klog n +k^2)))

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair <int,int> Pii;
#define Mod1(x) ((x>=P)&&(x-=P))
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(int i=a,i##end=b;i>=i##end;--i)

char IO;
template <class T=int> T rd(){
	T s=0; int f=0;
	while(!isdigit(IO=getchar())) if(IO=='-') f=1;
	do s=(s<<1)+(s<<3)+(IO^'0');
	while(isdigit(IO=getchar()));
	return f?-s:s;
}

const int N=2e5+10,P=1e9+7;

int n,k;
Pii A[N];
int C[11][11];
int s[N<<2][11],res[11];
int t[N<<2];
void Up(int p) {
	rep(i,0,k) s[p][i]=s[p<<1][i]+s[p<<1|1][i],Mod1(s[p][i]);
}
void Down(int p) {
	if(t[p]==1) return;
	t[p<<1]=1ll*t[p<<1]*t[p]%P;
	t[p<<1|1]=1ll*t[p<<1|1]*t[p]%P;
	rep(i,0,k) {
		s[p<<1][i]=1ll*s[p<<1][i]*t[p]%P;
		s[p<<1|1][i]=1ll*s[p<<1|1][i]*t[p]%P;
	}
	t[p]=1;
}

void Que(int p,int l,int r,int ql,int qr) {
	if(ql<=l && r<=qr) {
		rep(i,0,k) res[i]+=s[p][i],Mod1(res[i]);
		return;
	}
	Down(p);
	int mid=(l+r)>>1;
	if(ql<=mid) Que(p<<1,l,mid,ql,qr);
	if(qr>mid) Que(p<<1|1,mid+1,r,ql,qr);
}

void Upd(int p,int l,int r,int ql,int qr) {
	if(ql<=l && r<=qr) {
		rep(i,0,k) s[p][i]*=2,Mod1(s[p][i]);
		t[p]*=2,Mod1(t[p]);
		return;
	}
	Down(p);
	int mid=(l+r)>>1;
	if(ql<=mid) Upd(p<<1,l,mid,ql,qr);
	if(qr>mid) Upd(p<<1|1,mid+1,r,ql,qr);
	Up(p);
}
void Upd(int p,int l,int r,int x){
	if(l==r){
		rep(i,0,k) s[p][i]+=res[i],Mod1(s[p][i]);
		return;
	}
	Down(p);
	int mid=(l+r)>>1;
	x<=mid?Upd(p<<1,l,mid,x):Upd(p<<1|1,mid+1,r,x);
	Up(p);
}


int main(){
	n=rd(),k=rd();
	rep(i,1,n) A[i].first=rd(),A[i].second=rd();
	sort(A+1,A+n+1);
	rep(i,1,N*4-1) t[i]=1;
	rep(i,0,k) rep(j,C[i][0]=1,i) C[i][j]=(C[i-1][j-1]+C[i-1][j])%P;
	
	res[0]=1;
	Upd(1,0,n*2,0);

	rep(t,1,n) {
		int l=A[t].first,r=A[t].second;
		Upd(1,0,n*2,r,n*2);
		
		memset(res,0,sizeof res);
		Que(1,0,n*2,0,l-1);
		drep(i,k,0) rep(j,0,i-1) res[i]=(res[i]+1ll*C[i][j]*res[j])%P;
		Upd(1,0,n*2,r);

		memset(res,0,sizeof res);
		Que(1,0,n*2,l,r-1);
		Upd(1,0,n*2,r);
		
	}
	printf("%d
",s[1][k]);
}

原文地址:https://www.cnblogs.com/chasedeath/p/13729352.html