1343C. Alternating Subsequence

Posted on Edited on In

C. Alternating Subsequence

Description

输入 T(≤1e4) 表示 T 组数据。所有数据的 n 之和 ≤2e5。
每组数据输入 n(1≤n≤2e5) 和长为 n 的数组 a(-1e9≤a[i]≤1e9 且 a[i]≠0)。

选一个 a 的子序列，要求:

  1. 子序列是交替的（相邻元素一正一负）。
  2. 在满足 1 的前提下，子序列尽量长。
  3. 在满足 2 的前提下，子序列元素和尽量大。
    输出元素和的最大值。

注：子序列不一定连续。

Hints/Notes

  • 分组循环

Solution

Language: C++

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
#include <iostream>
#include <utility>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <unordered_map>
#include <unordered_set>
#include <algorithm>
#include <numeric>

#include <fstream>

using namespace std;

istream &in = cin;
ostream &out = cout;

void solve(vector<int>& nums) {
    long long pos_size = 0, pos_res = 0;
    for (int i = 0; i < nums.size(); ) {
        if (nums[i] < 0) {
            i++;
            continue;
        }
        int tmp = INT_MIN;
        pos_size++;
        for (; i < nums.size() && nums[i] > 0; i++) {
            tmp = max(tmp, nums[i]);
        }
        pos_res += tmp;
        tmp = INT_MIN;
        for (; i < nums.size() && nums[i] < 0; i++) {
            tmp = max(tmp, nums[i]);
        }
        if (tmp != INT_MIN) {
            pos_size++;
            pos_res += tmp;
        }
    }

    long long neg_size = 0, neg_res = 0;
    for (int i = 0; i < nums.size(); ) {
        if (nums[i] > 0) {
            i++;
            continue;
        }
        int tmp = INT_MIN;
        neg_size++;
        for (; i < nums.size() && nums[i] < 0; i++) {
            tmp = max(tmp, nums[i]);
        }
        neg_res += tmp;
        tmp = INT_MIN;
        for (; i < nums.size() && nums[i] > 0; i++) {
            tmp = max(tmp, nums[i]);
        }
        if (tmp != INT_MIN)  {
            neg_size++;
            neg_res += tmp;
        }
    }
    if (pos_size > neg_size) {
        out << pos_res << endl;
    } else if (pos_size < neg_size) {
        out << neg_res << endl;
    } else {
        out << max(pos_res, neg_res) << endl;
    }

}

int main() {
    int tc;
    in >> tc;
    for (int t = 1; t <= tc; t++) {
        int size;
        in >> size;
        vector<int> nums;
        for (int i = 0; i < size; i++) {
            int num;
            in >> num;
            nums.push_back(num);
        }
        solve(nums);
    }
    return 0;
}