ZJOI2018 胖 二分 ST表

原文链接https://www.cnblogs.com/zhouzhendong/p/ZJOI2018Day2T2.html

题目传送门 - BZOJ5308

题目传送门 - LOJ2529

题目传送门 - 洛谷P4501

题意

　　略。

题解

　　首先这个题目名称用来形容 cly 太好了。

　　考虑每一对 $(a_i,l_i)$ 对于答案的贡献。

　　我们可以发现每一条这种路径能够更新的节点是连续的一段。于是我们考虑二分边界。

　　设 x 为当前节点，我们当前二分到的节点为 y ，如果 x 不能更新节点 y ，那么，在区间 $[x,2y-x]$ 中必然存在一个点到 y 的最短路小于等于 x 到 y 的最短路。

　　于是，我们可以利用差分思想维护两个 ST 表来分别得到两边的最短路。

　　但是有一个特殊情况：

　　一个点同时被左右更新。

　　只需要特判就好了。

　　这题细节好多啊。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
int read(){
	int x=0;
	char ch=getchar();
	while (!isdigit(ch))
		ch=getchar();
	while (isdigit(ch))
		x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return x;
}
const int N=200005;
const LL INF=1LL<<57;
int n,m,w[N],Log[N];
LL x[N],s1[N][18],s2[N][18];
struct Node{
	int a,L;
}v[N];
bool cmp(Node A,Node B){
	return A.a<B.a;
}
LL query1(int L,int R){
	if (L>R)
		return INF;
	int d=Log[R-L+1];
	return min(s1[L+(1<<d)-1][d],s1[R][d]);
}
LL query2(int L,int R){
	if (L>R)
		return INF;
	int d=Log[R-L+1];
	return min(s2[L+(1<<d)-1][d],s2[R][d]);
}
int p[N];
LL l[N];
int main(){
	Log[1]=0;
	for (int i=2;i<N;i++)
		Log[i]=Log[i>>1]+1;
	n=read(),m=read();
	for (int i=1;i<n;i++)
		w[i]=read(),x[i+1]=x[i]+w[i];
	while (m--){
		int k=read();
		for (int i=1;i<=k;i++)
			v[i].a=read(),v[i].L=read();
		sort(v+1,v+k+1,cmp);
		for (int i=1;i<=k;i++)
			p[i]=v[i].a,l[i]=v[i].L;
		for (int i=1;i<=k;i++){
			s1[i][0]=l[i]-x[p[i]];
			s2[i][0]=l[i]+x[p[i]];
			for (int j=1;j<18;j++){
				s1[i][j]=s1[i][j-1];
				s2[i][j]=s2[i][j-1];
				if (i-(1<<(j-1))>0){
					s1[i][j]=min(s1[i][j],s1[i-(1<<(j-1))][j-1]);
					s2[i][j]=min(s2[i][j],s2[i-(1<<(j-1))][j-1]);
				}
			}
		}
		LL ans=0;
		for (int i=1;i<=k;i++){
			int now=p[i],R=now,L=now;
			for (int j=17;j>=0;j--){
				int t=R+(1<<j);
				if (t>n)
					continue;
				int pL=i+1;
				int pM1=upper_bound(p+1,p+k+1,t)-p-1;
				int pM2=lower_bound(p+1,p+k+1,t)-p;
				int pR=upper_bound(p+1,p+k+1,t*2-now)-p-1;
				if (query1(pL,pM1)<=l[i]-x[now])
					continue;
				if (query2(pM2,pR)-x[t]<=l[i]+x[t]-x[now])
					continue;
				R=t;
			}
			for (int j=17;j>=0;j--){
				int t=L-(1<<j);
				if (t<=0)
					continue;
				int pR=i-1;
				int pM1=lower_bound(p+1,p+k+1,t)-p;
				int pM2=upper_bound(p+1,p+k+1,t)-p-1;
				int pL=lower_bound(p+1,p+k+1,t*2-now)-p;
				if (query2(pM1,pR)<=l[i]+x[now])
					continue;
				if (query1(pL,pM2)+x[t]<=l[i]+x[now]-x[t])
					continue;
				L=t;
			}
			ans+=R-L+1;
			if (R<n){
				R++;
				int pR=lower_bound(p+1,p+k+1,R*2-now)-p;
				if (p[pR]==R*2-now){
					int pM1=upper_bound(p+1,p+k+1,R)-p-1;
					int pM2=lower_bound(p+1,p+k+1,R)-p;
					if (query1(i+1,pM1)>l[i]-x[now]
					  &&query2(pM2,pR-1)-x[R]>l[i]+x[R]-x[now]
					  &&l[i]+x[R]-x[now]==l[pR]+x[p[pR]]-x[R])
					  	ans++;
				}
			}
		}
		printf("%lld\n",ans);
	}
	return 0;
}