P5979 [PA2014] Druzyny 题解

对于一个固定的右端点 \(r\),左端点 \(l\) 合法当且仅当 \(\max(d_l,d_{l+1},\dots d_r)\le r-l+1 \le\min(c_l,c_{l+1},\dots,c_r)\)

容易得到一个很朴素的 dp:记 \(f_i\) 表示前 \(i\) 个位置可以分成的组的数目的最大值,\(g_i\) 表示有多少种分组方案能达到最大值,直接枚举左端点转移,时间复杂度 \(\mathcal O(n^2)\)。转移点不是连续的,不好直接优化。

考虑对这个 dp 进行一个分治,对于每个分治的区间 \([l,r]\) 处理 \(f_{[l,\text{mid}]},g_{[l,\text{mid}]}\)\(f_{[\text{mid}+1,r]},g_{[\text{mid}+1,r]}\) 的贡献。发现 \(c\) 对每个右端点的限制是只能取某个后缀的左端点,可以直接二分出这个后缀,接下来只要考虑 \(d\) 的限制。

很容易想到倒着枚举 \([l,\text{mid}]\) 中的每个位置,维护 \(d\) 的最大值,可以得到一个右端点的位置 \(\ge \max(d_{[l,\text{mid}]})+l-1\) 的限制。考虑对 \([\text{mid}+1,r]\) 每个位置建一个桶,在这个限制的对应位置处插入这个左端点。接下来正着枚举 \([\text{mid}+1,r]\) 中的每个位置,把桶内的位置插入线段树。对于每个右端点也可以得到一个类似的限制,在线段树上查询即可,最后更新一下就做完了。

时间复杂度 \(\mathcal O(n\log^2n)\),参考代码:

#include<bits/stdc++.h>
#define ll long long
#define mxn 1000003
#define md 1000000007
#define pb push_back
#define rep(i,a,b) for(int i=a;i<=b;++i)
#define rept(i,a,b) for(int i=a;i<b;++i)
#define drep(i,a,b) for(int i=a;i>=b;--i)
using namespace std;
struct node{
	ll a,c;
}f[mxn],t[mxn<<2];
int n,lg[mxn],ps[mxn],a[mxn],b[20][mxn];
vector<int>s[mxn];
const int INF=1e9;
node operator+(node x,node y){
	return {max(x.a,y.a),((x.a>=y.a?x.c:0)+(y.a>=x.a?y.c:0))%md};
}
int ask2(int l,int r){
	int k=lg[r-l+1];
	return min(b[k][l],b[k][r-(1<<k)+1]);
}
void change(int p,int x,node y,int l,int r){
	if(l==r){t[p]=y;return;}
	int mid=(l+r)>>1;
	if(x<=mid)change(p<<1,x,y,l,mid);
	else change(p<<1|1,x,y,mid+1,r);
	t[p]=t[p<<1]+t[p<<1|1];
}
node ask(int p,int l,int r,int L,int R){
	if(l<=L&&R<=r)return t[p];
	int mid=(L+R)>>1;
	if(l<=mid&&r>mid)return ask(p<<1,l,r,L,mid)+ask(p<<1|1,l,r,mid+1,R);
	if(l<=mid)return ask(p<<1,l,r,L,mid);
	return ask(p<<1|1,l,r,mid+1,R);
}
void build(int p,int l,int r){
	t[p]={-INF,0};
	if(l==r)return;
	int mid=(l+r)>>1;
	build(p<<1,l,mid);build(p<<1|1,mid+1,r);
}
void solve(int l,int r){
	if(l==r)return;
	int mid=(l+r)>>1;
	solve(l,mid);
	rep(i,mid+1,r)s[i].clear();
	int mx=0;
	drep(i,mid,l){
		if(mx+i<=r)s[max(mx+i,mid+1)].pb(i);
		mx=max(mx,a[i]);
	}
	mx=0;
	rep(i,mid+1,r){
		mx=max(mx,a[i]);
		for(int j:s[i])change(1,j,{f[j].a+1,f[j].c},0,n);
		int L=max(ps[i]-1,l),R=min(i-mx,mid);
		if(L<=R)f[i]=f[i]+ask(1,L,R,0,n);
	}
	rep(i,l,mid)change(1,i,{-INF,0},0,n);
	solve(mid+1,r);
}
signed main(){
	scanf("%d",&n);
	rep(i,2,n)lg[i]=lg[i>>1]+1;
	rep(i,1,n)scanf("%d%d",&a[i],&b[0][i]);
	rept(k,1,20){
		rep(i,1,n-(1<<k)+1){
			b[k][i]=min(b[k-1][i],b[k-1][i+(1<<(k-1))]);
		}
	}
	rep(i,1,n){
		int l=1,r=i;
		while(l<r){
			int mid=(l+r)>>1;
			if(ask2(mid,i)>=i-mid+1)r=mid;
			else l=mid+1;
		}
		ps[i]=l;
	}
	f[0].c=1;
	rep(i,1,n)f[i].a=-INF;
	build(1,0,n);
	solve(0,n);
	if(f[n].a<=0)puts("NIE");
	else cout<<f[n].a<<" "<<f[n].c;
	return 0;
}
posted @ 2024-09-19 23:39  zifanwang  阅读(4)  评论(0编辑  收藏  举报