思路:枚举每个t,求出对应的满足条件的s。
代码:
#includeusing namespace std;#define ll long long#define pb push_back#define mem(a,b) memset((a),(b),sizeof(a))#define pii pair #define mp make_pairconst int N=1e5+5;int a[N];int c1[N]={ 0};int c2[N]={ 0};pii ans[N];int main(){ ios::sync_with_stdio(false); cin.tie(0); int n; cin>>n; for(int i=1;i<=n;i++) { cin>>a[i]; c1[i]=c1[i-1]; c2[i]=c2[i-1]; if(a[i]==1)c1[i]++; else c2[i]++; } int s1,s2,last1,last2,c=0; for(int t=1;t<=n;t++) { s1=s2=last1=last2=0; while(true) { int t1=lower_bound(c1+1,c1+n+1,last1+t)-c1; int t2=lower_bound(c2+1,c2+n+1,last2+t)-c2; if(t1>n&&t2>n)break; int mn=min(t1,t2); if(mn==t1)s1++; else s2++; if(mn==n) { if(s2>s1&&a[n]==2||s1>s2&&a[n]==1) { ans[c].first=max(s1,s2); ans[c++].second=t; } break; } last1=c1[mn];//上一局的最后得分 last2=c2[mn];//上一局的最后得分 } } cout< <