整理的算法模板:ACM算法模板总结(分类详细版)
题目链接:https://codeforces.ml/contest/1330
题意:一个数组分成两段,长度分别为l1,l2;每段里面分别包括1~l1或者1~l2的所有数字;求有多少种l1,l2符合要求;
思路:
满足以下两个条件即可
- 段的最大值maxx等于段的长度
- 段内没有重复元素
实现方法有很多种,我刚开始写就麻烦了,用前缀和以及后缀和也可以做;
预处理数组的前缀和,后缀和;然后判断该maxx段的和是否等于用等差数列求出来的和;正着判断一次,反着判断一次即可;
注意l1==l2的情况需要特判;(这样写还是好麻烦,我觉得还会有很简单的思想,在线等.....)
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=200005;
ll a[N],inv[N],env[N];
ll get(ll x)
{
return x*(1+x)/2;
}
int main()
{
int t;
cin >>t;
while(t--)
{
int n,cnt=0;
cin >>n;
memset(inv,0,sizeof inv);
memset(env,0,sizeof env);
ll maxx=-1;
for(int i=1;i<=n;i++) cin >>a[i],maxx=max(maxx,a[i]),inv[i]=inv[i-1]+a[i];
for(int i=n;i>=1;i--) env[i]=env[i+1]+a[i];
if(maxx==n)
{
cout <<0<<endl;
continue;
}
ll l=maxx,r=n-maxx,x=0,y=0;
if(inv[maxx]==get(maxx)&&env[maxx+1]==get(n-maxx)) x++,cnt++;
if(env[n-maxx+1]==get(maxx)&&inv[n-maxx]==get(n-maxx)) y++,cnt++;
if(x&&y&&l==r) cnt--,x--;
cout <<cnt<<endl;
if(x) cout <<l<<" "<<r<<endl;
if(y) cout <<r<<" "<<l<<endl;
}
}