●CodeForces 549F Yura and Developers

题链：

http://codeforces.com/problemset/problem/549/F
题解：

分治，链表。
考虑对于一个区间[L,R],其最大值在p位置，
那么答案的贡献就可以分为3部分：
1.[L,p-1]中合法的区间
2.[p+1,R]中合法的区间
3.[L,R]中经过p的合法区间。
前两个因为是相同子问题，可以递归解决。
考虑如何求出3.的答案。
令S[i]表示$\sum_{k=1}^{i}A[i]$(即前缀和)
显然对于一组(l,r)表示的区间[l,r]，
当且仅当满足L<=l<=p且p<r<=R(或L<=l<p且p<=r<=R)，
并且 S[r]-S[l-1]-A[p]≡0 (mod K)时，才是合法区间。

代码：

#include<bits/stdc++.h>
#define MAXN 300050
#define INF 0x3f3f3f3f
using namespace std;
int N,K,more;
int A[MAXN],S[MAXN],L[MAXN],R[MAXN];
struct LINK{
	int lnt;
	int nxt[MAXN*20],val[MAXN*20],sign[MAXN*20],head[MAXN];
	LINK(){lnt=2;}
	void Add(int u,int v,int s){
		if(u<0) return;
		val[lnt]=v; sign[lnt]=s; nxt[lnt]=head[u]; head[u]=lnt++;
	}
	void Query(int l,int r,int v){
		Add(l-1,v,-1); Add(r,v,1);
	}
	long long Getans(){
		long long ret=0;
		static int cnt[1000050];
		for(int i=0;i<=N;i++){
			cnt[S[i]]++;
			for(int j=head[i];j;j=nxt[j])
				ret+=sign[j]*cnt[val[j]];
		}
		return ret;
	}
}Q;
void prework(){
	static int stk[MAXN],top;
	A[0]=A[N+1]=INF;
	stk[top=1]=0;
	for(int i=1;i<=N;i++){
		while(A[stk[top]]<=A[i]) top--;
		L[i]=stk[top]+1; stk[++top]=i;
	}
	stk[top=1]=N+1;
	for(int i=N;i>=1;i--){
		while(A[stk[top]]<A[i]) top--;
		R[i]=stk[top]-1; stk[++top]=i;
	}
	for(int i=1;i<=N;i++)
		S[i]=(S[i-1]+A[i])%K;
}
void solve(){
	int al,ar,qv,ql,qr,sign;
	for(int i=1;i<=N;i++){
		if(L[i]==R[i]) continue;
		more++;
		if(i-L[i]+1<=R[i]-i+1) al=L[i]-1,ar=i-1,ql=i,qr=R[i],sign=1;
		else al=i,ar=R[i],ql=L[i]-1,qr=i-1,sign=-1;
		for(int j=al;j<=ar;j++){
			qv=(1ll*S[j]+sign*A[i]%K+K)%K;
			Q.Query(ql,qr,qv);
		}
	}
}
int main(){
	scanf("%d%d",&N,&K);
	for(int i=1;i<=N;i++) scanf("%d",&A[i]);
	prework();
	solve();
	long long ans=Q.Getans();
	printf("%lld\n",ans-more);
	return 0;
}

原文地址：https://www.cnblogs.com/zj75211/p/8541888.html