Codeforces1364B Most socially-distanced subsequence (思维)

题目链接: Most socially-distanced subsequence

大致题意

给出一个长度为n的序列p, 你需要找到p的一个子序列s, s的长度至少为2, 且|s₁ - s₂| + |s₂ - s₃| + … + |s_k-1 - s_k|尽可能大, 如果多个子序列的值相同, 输出任一长度最短的子序列.

解题思路

考虑到对于上式|s₁ - s₂| + |s₂ - s₃| + … + |s_k-1 - s_k|进行化简:

①假设当前序列单调递增, 则可化简为: (s₂ - s₁) + (s₃ - s₂) + … + (s_k - s_k-1) = sk - s1
②假设当前序列单调递减, 则可化简为: (s₁ - s₂) + (s₂ - s₃) + … + (s_k-1 - s_k) = s1 - sk

我们发现, 如果对于一段单调递增 或 单调递减的区间[l, r], 只有p[l]和p[r]是有意义的, 我们选择其他的元素并不会使得当前序列的值增加或减少.

对于一段序列, 我们可以将其拆分成若干个互不相交的 递增区间 和 递减区间. 我们需要选择每个相应区间的开头元素和结尾元素. 这其实相当于我们选择波峰元素和波谷元素以及p[1] 和 p[n].

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1E5 + 10;
int a[N];
int main()
{
	int t; cin >> t;
	while (t--) {
		vector<int> v;

		int n; scanf("%d", &n);
		rep(i, n) scanf("%d", &a[i]);

		rep(i, n) {
			if (i == 1 or i == n) v.push_back(a[i]);
			else if (a[i - 1] < a[i] and a[i + 1] < a[i]) v.push_back(a[i]);
			else if (a[i - 1] > a[i] and a[i + 1] > a[i]) v.push_back(a[i]);
		}

		printf("%u\n", v.size());
		for (auto& op : v) printf("%d ", op); puts("");
	}
    
	return 0;
}

END