程序员

636div3 - C. Alternating Subsequ

2020-05-24  本文已影响0人  waaagh

首先想到二维dp,不过肯定超时。
先找规律:同号的为一类,每类找最大的,相加即为所求。比如:(1 2 3) (-1 -2),最大序列数肯定为2,和即为最大的之和3+(-1)=2。算法使用two-pointer,两指针之间为一类数。
代码:

#include<bits/stdc++.h>
using namespace std;

const int N = 2e5+10;
using ll = long long;

int t, n;
ll a[N];
inline int sgn(int x) {
    return x>0?1:-1;
}
int main() {
    cin >> t;
    while(t--) {
        cin >> n;
        for(int i=1; i<=n; i++) {
            cin >> a[i];
        }
        int i, j;
        ll ans = 0;
        for(i=1; i<=n; i++) {
            ll cur = a[i];
            j = i;
            while(j<=n && sgn(a[i]) == sgn(a[j])) {
                cur = max(cur, a[j]);
                j++;
            }
            ans += cur;
            i = j-1;
        }
        cout << ans << endl;
    }
    return 0;
}
上一篇下一篇

猜你喜欢

热点阅读