Looksery Cup 2015 F. Yura and Developers（单调栈+二分+分治）（难*）

题目链接


题意：给定一个数组，问有多少区间满足：去掉最大值之后，和是k的倍数。
思路：日后补。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+2;
stack<int>s;
ll ans,sum[maxn],l[maxn],r[maxn];
vector<int>v[maxn];
int a[maxn];
int query(ll x,int l,int r)
{
	return upper_bound(v[x].begin(),v[x].end(),r)-lower_bound(v[x].begin(),v[x].end(),l);
}
void slove()
{
	int n,k; 
	scanf("%d %d",&n,&k);
	v[0].push_back(0);
	for(int i=1;i<=n;++i) {
		scanf("%d",&a[i]);
		sum[i]=(sum[i-1]+a[i])%k;
		v[sum[i]].push_back(i);
	}
	a[n+1]=1e9+7;
	s.push(0);
	for(int i=1;i<=n;++i)
	{
		while(s.size()>1&&a[s.top()]<a[i]) s.pop();
		l[i]=s.top();
		s.push(i);
	}
	while(!s.empty()) s.pop();
	s.push(n+1);
	for(int i=n;i>=1;--i)
	{
		while(s.size()>1&&a[s.top()]<=a[i]) s.pop();
		r[i]=s.top();
		s.push(i);
	}
	for(int i=1;i<=n;++i) a[i]%=k;
	for(int i=1;i<=n;++i)
	if(i-l[i]<r[i]-i)
	for(int j=l[i]+1;j<=i;++j) ans+=query((sum[j-1]+a[i])%k,i,r[i]-1);
	else for(int j=i;j<r[i];++j) ans+=query((sum[j]-a[i]+k)%k,l[i],i-1);
	printf("%lld\n",ans-n);
}
int main()
{
	slove();
}