比赛 2026.8.28 评测结果 AAAAAAAAAA
题目名称 一周一次买下同班同学的那些事 最终得分 100
用户昵称 123 运行时间 0.316 s
代码语言 C++ 内存使用 7.16 MiB
提交时间 2026-08-28 09:23:18
显示代码纯文本
#include <bits/stdc++.h>
#define pb push_back 
using namespace std;
const int N=4e5+10;
const long long inf=-1e18;
int n,a[N],f[N][2][2],to[N][2][2];
long long dp[N][2][2];
vector<int> ans,v,p;
set<int> s;
void solve()
{
    cin>>n;ans.clear(),v.clear();
    for (int i=1;i<=n;i++) cin>>a[i];
    dp[n+1][0][0]=0,dp[n+1][1][0]=dp[n+1][0][1]=dp[n+1][1][1]=-1e18;
    for (int i=n;i>=1;i--)
    {
        dp[i][0][1]=dp[i][1][0]=dp[i][1][1]=-1e18;
        dp[i][0][0]=dp[i+1][0][0]+a[i];
        f[i][0][0]=0,to[i][0][0]=0;
        for (int j=0;j<2;j++)
        {
            long long cnt1=dp[i+1][j][1]+(j ? -a[i] : a[i]),cnt2=dp[i+1][j^1][1]+(j ? -a[i] : a[i]),cnt3=((a[i]>0 && j==1) ? dp[i+1][0][0]-a[i] : inf);
//            cout<<i<<" "<<j<<" "<<cnt1<<" "<<cnt2<<" "<<cnt3<<endl;
            if (cnt1>cnt2 && cnt1>cnt3) f[i][j][1]=0,dp[i][j][1]=cnt1,to[i][j][1]=1;
            else if (cnt2>cnt3) f[i][j][1]=1,dp[i][j][1]=cnt2,to[i][j][1]=1;
            else f[i][j][1]=1,dp[i][j][1]=cnt3,to[i][j][1]=0;
        }
    }
    int now=(max(dp[1][0][1],dp[1][0][0])>=dp[1][1][1] ? 0 : 1),x=(now==0 && dp[1][0][0]>dp[1][0][1] ? 0 : 1);
    for (int i=1;i<=n;i++)
    {
        if (f[i][now][x]) v.pb(i),x=to[i][now][x],now^=1;
    }
    for (auto &j:v)
    {
        if (a[j]>0)
        {
            ans.pb(j);
            for (auto &k:p) ans.pb(k);
            p.clear();
        }
        else p.pb(j); 
    }
    cout<<ans.size()<<"\n";
    for (auto &j:ans) cout<<j<<" ";
    cout<<"\n";
}
int main() {
    freopen("bought.in","r",stdin);
    freopen("bought.out","w",stdout);
    ios::sync_with_stdio(0),cin.tie(0);
    int t;
    cin>>t;
    while (t--) solve();
    return 0;
}