P11157 【MX-X6-T3】さよならワンダーランド

原题链接

Solution

对于这样的题，不急着做，先简化一下式子，令 $k=i+j$，则我们就是要在 $[1,n]$ 中找一个 $k$ 满足

$a_i<=j<=a_k$，而 $j=k-i$，所以带入并移项可得到
$a_i+i<=k<=a_k+i$，所以我们要在 $[a_i+i,n]$ 中找一个 $k$，使 $k-a_k<=i$，若 $a_i+i>n$ 则直接无解。

由此可以想到 st 表，一个下标为 $i$ 的数权值为 $i-a_i$，维护最小值下标，最后这道题就做完了。

---

Code

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;

int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){
		if(ch=='-') f=-1;
		ch=getchar();
	}
	while(isdigit(ch)) x=x*10+ch-'0',ch=getchar();
	return x*f;
}

int n,a[N],f[N][20];

int cmp(int x,int y){
	return (x-a[x]<y-a[y])?x:y;
}

int query(int x,int y){
	int k=log2(y-x+1);
	return cmp(f[x][k],f[y-(1<<k)+1][k]);
}

int main(){
	
	n=read();
	for(int i=1;i<=n;i++) a[i]=read(),f[i][0]=i;
	for(int i=1;(1<<i)<=n;i++)
		for(int j=1;j<=n-(1<<i)+1;j++)
			f[j][i]=cmp(f[j][i-1],f[j+(1<<(i-1))][i-1]);
				
	for(int i=1;i<=n;i++){
		int l=max(1,a[i]+i),r=n;
		if(l>r) cout<<0<<endl;
		else{
			int k=query(l,r);
			if(k>a[k]+i) cout<<0<<endl;
			else cout<<1<<" "<<k-i<<endl;
		}
	}
	
	return 0;
}