C2. Magnitude (Hard Version)

原題連結

題解

由於使用操作二會讓負數變成正數，所以我們考慮讓操作二在c最小且為負數的點使用
在使用完操作二之後，之後的c肯定非負，所以在此之後兩種操作都可以使用

實施

先判斷存不存在c最小且為負數的點，然後再統計所有c最小且為負數的點的貢獻

code

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

typedef long long ll;
const int MAXN = 200005;
const ll mod = 998244353;

ll a[MAXN];
ll pre[MAXN] = {0};

// 快速冪計算
ll qpow(ll base, ll exp) {
    ll result = 1;
    base %= mod; // 取模
    while (exp > 0) {
        if (exp % 2 == 1)
            result = (result * base) % mod; // 取模
        base = (base * base) % mod; // 取模
        exp /= 2;
    }
    return result;
}

int main() {
    int t;
    cin >> t;
    while (t--) {
        int n;
        cin >> n;
        for (int i = 1; i <= n; i++)
        {
            cin >> a[i];
            pre[i] = pre[i-1] + a[i];
        }

        int pos = 1;
        for (int i = 2; i <= n; i++) {
            if (pre[i] < pre[pos]) pos = i;
        }

        if(pre[pos]>=0)
        {
            cout<<qpow(2,n)%mod<<endl;
            continue;
        }
        ll ans=0,tem=1;
        for(int i=1;i<=n;i++)
        {
            if(pre[i]>=0) tem<<=1;
            tem%=mod;
            if(pre[i]==pre[pos]) ans+=tem*qpow(2,n-i)%mod;
            ans%=mod;
        }

        cout << ans % mod << endl; // 結果取模
    }
    return 0;
}