「2019牛客多校第三场 G」Removing Stones【分治】

题目链接：https://ac.nowcoder.com/acm/contest/883/G

题意

• 就是给你一个数组，问有多少个区间$[l,r]$，使得可以经过任意次操作使得区间内的所有数变为$0$，操作定义为每次可以选两个数，同时将他们减一，显然如果区间和为奇数，那么不可能清空区间，但题目定义如果区间和为奇数，先将区间最小的那个数减一

题解

• 先可以从最坏的情况考虑，假如最大的那个数大于其他所有数的和，显然不能清空，所以要求就是最大的数小于等于其他所有数的和
• 然后就可以想到套路操作，类比CF1156E，就容易想到分治处理，即对于每个区间，先$rmq$找到最大值位置，记为$max\_pos$，先处理跨过$max\_pos$的所有区间数量，然后递归处理区间$[l,max\_pos-1]$和区间$[max\_pos+1,r]$的贡献
• 那么怎么处理跨过$max\_pos$的情况呢？由第一条结论可知，区间和必须大于等于二倍区间最大值，所以可以在$[max\_pos,r]$内$lower\_bound$一下就行了
• 算法复杂度$O(n(\log n{)}^2)$

代码

#include<bits/stdc++.h>

using namespace std;
const int maxn=3e5+10;

int a[maxn],n,dp[maxn][25],pos[maxn][25];
long long pre[maxn],suf[maxn];

void init()
{
	for(int i=1;i<=n;i++) dp[i][0]=a[i],pos[i][0]=i;
	for(int j=1;(1<<j)<=n;j++) for(int i=1;i+(1<<j)-1<=n;i++) {
		if(dp[i][j-1]>=dp[i+(1<<(j-1))][j-1]) dp[i][j]=dp[i][j-1],pos[i][j]=pos[i][j-1];
		else dp[i][j]=dp[i+(1<<(j-1))][j-1],pos[i][j]=pos[i+(1<<(j-1))][j-1];
	}
}

int query(int l,int r)
{
	int k=0;
	while(l+(1<<(k+1))-1<=r) k++;
	if(dp[l][k]>=dp[r-(1<<k)+1][k]) return pos[l][k];
	return pos[r-(1<<k)+1][k];
}

long long dfs(int l,int r)
{
	if(l>=r) return 0;int max_pos=query(l,r);long long res=0;
	if(max_pos-l<r-max_pos) for(int i=l;i<=max_pos;i++) res+=r-(lower_bound(pre+max_pos,pre+r+1,pre[i-1]+2*a[max_pos])-pre-1);
	else for(int i=max_pos;i<=r;i++) res+=(n-l+1)-(lower_bound(suf+n-max_pos+1,suf+n-l+2,suf[n-i]+2*a[max_pos])-suf-1);
	return res+dfs(l,max_pos-1)+dfs(max_pos+1,r);
}

int main()
{
	int t;scanf("%d",&t);
	while(t--) {
		scanf("%d",&n);
		for(int i=1;i<=n;i++) scanf("%d",&a[i]);
		for(int i=1;i<=n;i++) pre[i]=pre[i-1]+a[i],suf[i]=suf[i-1]+a[n-i+1];
		init();
		printf("%lld\n",dfs(1,n));
	}
}