Codeforces B. Most socially-distanced subsequence (构造 / 模拟) (Round #649 Div.2)

传送门

题意： 给出一个长度为n的序列a，找到其长度至少为2的子序列(不一定连续)。若子序列长度为k，要求子序列的特定值：|a[1] - a[2]| + |a[2] - a[3]| + …… + |a[k - 1] - a[k]|最大，且长度k最小。

思路：

• 由于需要求的特定值是子序列的连续差之和，所有我们可以O(n)遍历一遍，将不符合条件的元素踢掉即可。
• 单独处理n == 2的情况。然后将a[1]和a[2]加入答案组，并用两个指针i，j分别指向索引1和2.
• 用指针k从3开始遍历序列a：若i，j，k的小序列的特定值比i，k的特定值大，说明j元素符合条件；
• 反之就需要踢掉元素a[j]再将元素a[k]加入答案组。最后还需要将i，j指针向后移动一位。

代码实现：

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {{1, 0}, {-1, 0}, {0, 1}, {0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;

int t, n, a[N];
vector<int> vt;

signed main()
{
    IOS;

    cin >> t;
    while(t --){
        cin >> n;
        vt.clear();
        for(int i = 1; i <= n; i ++)
            cin >> a[i];
        if(n == 2){
            if(a[1] == a[2])
                cout << 1 << endl << a[1] << endl;
            else cout << 2 << endl << a[1] << " " << a[2] << endl;
            continue;
        }
        vt.push_back(a[1]);
        vt.push_back(a[2]);
        int i = 1, j = 2;
        for(int k = 3; k <= n; k ++){
            if(abs(a[i] - a[j]) + abs(a[j] - a[k]) <= abs(a[i] - a[k]))
                vt.pop_back();
            vt.push_back(a[k]);
            i = j; j = k;
        }
        cout << vt.size() << endl;
        for(auto it : vt) cout << it << " ";
        cout << endl;
    }

    return 0;
}