牛客练习赛14 B.区间的连续段(前缀和 倍增)

原题链接

题意：

给你一个长为$n$的序列$a$和一个常数$k$。有$m$次询问，每次查询一个区间$[l,r]$内所有数最少分成多少个连续段，使得每段的和都$<=k$。如果这一次查询无解，输出$"Chtholly"$。

思路：

• 先考虑简化版：给定一个序列和$k$，问将该序列最少分成多少个连续段，使得每段的和都$<=k$。
• 从前向后枚举，对于每次结果都贪心的考虑，如果当前和大于$k$就重新开一段。
• 也就是说需要处理的就是，对于每个点能够到达的最远的点。
• 回到这个题，类比的处理一下。但是加了区间限制之后，最远的点不一定在区间里，所以略微变形一下：维护$dp[i][j]$表示从点$i$开始，分成$2^j$段，最远能够到达的位置的下一个点。
• 由于$2^j=2^{j-1}+2^{j-1}$，所以$dp[i][j]=dp[dp[i][j-1]][j-1]$
• 对于$dp[i][0]=x$，找一个位置$[i,x-1]$使得该段的和恰好小于$k$;$sum[i]$表示$a$的前缀和。也就是说$sum[x-1]-sum[i-1]<k，sum[x-1]<sum[i-1]+k$，二分找最大的就可以。
• 对于查询：$"Chtholly"$的情况就是$a[i]>k,l<=i<=r$。用$tot[i]$表示前$i$个数中$>k$的数的个数，每次查询$tot[r]-tot[l-1]$，如果大于$0$，就是$"Chtholly"$
• 对于其他情况，从大到小枚举，类似二进制拆分的原理。$[l,st[l][i])$表示将区间分成了$2^i$段，所以答案要加$2^i$。
• 注意$dp$维护的是下一个位置，所以最后那段是不计算的，要额外再$+1$

代码：

const int maxn=1000000+7;

ll n,m,k,a[maxn],tot[maxn];
ll dp[maxn][25];
ll sum[maxn];
int main(){
	n=read,m=read,k=read;
	rep(i,1,n){
		a[i]=read;
		sum[i]=sum[i-1]+a[i];
		if(a[i]>k) tot[i]=tot[i-1]+1;
		else tot[i]=tot[i-1];
	}
	rep(i,1,n){
		dp[i][0]=upper_bound(sum+1,sum+1+n,sum[i-1]+k)-(sum+1)+1;
	}
	for(int j=1;(1<<j)<n;j++)
		for(int i=1;i+(1<<j)<=n;i++){
			dp[i][j]=dp[dp[i][j-1]][j-1];
		}
	while(m--){
		int l=read,r=read;
		if(tot[r]-tot[l-1]>0){
			puts("Chtholly");
		}
		else{
			int res=1;
			for(int i=20;dp[l][0]<=r;i--)
				if(dp[l][i]&&dp[l][i]<=r)
					l=dp[l][i],res+=(1<<i);
			cout<<res<<endl;
		}
	}
	return 0;
}