1. 程式人生 > >ZJOI2018 胖 二分 ST表

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;
}