CodeForces933A A Twisty Movement

题目链接

给出一个长度为n的由1和2组成的序列问翻转一次任意[l,r]区间后最长的不下降子序列长度

观察发现最后答案序列为1...2...1...2则枚举分界点即可

#include <cstdio>
#include <iostream>
using namespace std;
int N;
int A[2050];
int sum1[2050], sum2[2050];
int main() {
	scanf("%d", &N);
	for(int i = 1; i <= N; i++) {
		scanf("%d", &A[i]);
	}
	for(int i = 1; i <= N; i++) {
		if(A[i] == 1) sum1[i]++;
		else sum2[i]++;
		sum1[i] += sum1[i - 1];
		sum2[i] += sum2[i - 1];
	}
	int ans = 0;
	for(int i = 1; i <= N; i++) {
		int maxn1 = 0, maxn2 = 0;
		for(int j = 0; j <= i; j++) 
		maxn1 = max(maxn1, sum1[j] + sum2[i] - sum2[j]);
		for(int j = i; j <= N; j++) 
		maxn2 = max(maxn2, sum1[j] - sum1[i] + sum2[N] - sum2[j]);
		ans = max(ans, maxn1 + maxn2);
	}
	printf("%d
", ans);
	return 0;
}