제출 #668934

#제출 시각아이디문제언어결과실행 시간메모리
668934Dan4Life수열 (APIO14_sequence)C++17
22 / 100
2051 ms102240 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define all(a) a.begin(),a.end()
#define mp make_pair
#define fi first
#define se second

const int maxn = (int)2e2+10;
int n, K, a[maxn];
int pre[maxn];
int dp[maxn][maxn][201];
pair<int,int> p[maxn][maxn][201];
vector<int> v;
int sum(int i, int j) {return pre[j+1]-pre[i];}

int recur(int l, int r, int k){
    if(r-l+1<=k or !k) return 0;
    if(dp[l][r][k]!=-1) return dp[l][r][k];
    int ans = 0;
    int xx =l, yy=r-1;
    while(xx<yy){
        int mid = (xx+yy+1)/2;
        if(sum(l,mid)<=sum(mid+1,r)) xx=mid;
        else yy=mid-1;
    }

    for(int x = max(l,xx-15); x < min(r,xx+15); x++){
        for(int y = 0; y < min(k,x-l+1); y++){
            int tot = recur(l,x,y)+recur(x+1,r,k-y-1)+sum(l,x)*sum(x+1,r);
            if(ans < tot) ans=tot, p[l][r][k]=mp(x,y);
        }
    }
    return dp[l][r][k]=ans;
}

void recur2(int l, int r, int k){
    if(r-l+1<=k or !k) return;
    v.pb(p[l][r][k].fi);
    recur2(l,p[l][r][k].fi,p[l][r][k].se);
    recur2(p[l][r][k].fi+1,r,k-p[l][r][k].se-1);
}

int32_t main()
{
    ios_base::sync_with_stdio(false); cin.tie(0);
    cin >> n >> K; memset(dp,-1,sizeof(dp));
    for(int i = 0; i < n; i++) cin >> a[i], pre[i+1]=pre[i]+a[i];
    cout << recur(0,n-1,K) << "\n"; recur2(0,n-1,K);
    sort(all(v)); for(auto u : v) cout << u+1 << " ";
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...