제출 #855282

#제출 시각아이디문제언어결과실행 시간메모리
855282vjudge1수열 (APIO14_sequence)C++17
0 / 100
8 ms4700 KiB
// Bolatulu
#include <bits/stdc++.h>

typedef long long ll;
typedef unsigned long long ull;
typedef double db;
#define int long long
#define kanagattandirilmagandiktarinizdan ios_base::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
#define pb push_back
#define F first
#define S second
#define md (tl+tr)/2
#define TL v+v,tl,mid
#define TR v+v+1,mid+1,tr

#pragma GCC target( "sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#pragma GCC optimize("Ofast,unroll-loops,fast-math,O3")

using namespace std;

int binpow(int a,int n,int M) {
    if (n==0)
        return 1;
    if (n%2!=0)
        return (a * binpow(a,n-1,M))%M;
    int z=binpow(a,n/2,M);
    return (z*z)%M;
}

int rnd() {
    int x = rand() << 15;
    return rand() ^ x;
}

struct target {
    int l,r,x,y,ans;
};

bool cmp(target a, target b) {
    if (a.l==b.l)
        return a.r<b.r;
    return a.l<b.l;
}

const ll INF = 1e18+9;
const int N =  4e6+7;
const int M = 1e9+7;
const ll HZ = 1e5;
const int MAX = INT_MAX;
const int MIN = INT_MIN;
const db pi = 3.141592653;
const int P=31;

int n,k,a[N],ps[N];
bool u[N];

void solve() {
    cin >> n >> k;
    set <int> v1,v2;
    for (int i=1;i<=n;i++)
        cin >> a[i],ps[i]=ps[i-1]+a[i];
    v1.insert(0), v2.insert(n+1);
    int ansum=0;
    vector <int> ans;
    for (int i=1;i<=k;i++) {
        int mx=INF,pos,sum=0;
        for (int j=1;j<n;j++) {
            if (u[j])
                continue;
            auto f=v1.upper_bound(j);
            f--;
            int z= *v2.lower_bound(j+1), x= *f;
            if (z==j or x==j)
                continue;
            if (abs((ps[j]-ps[x])-(ps[z-1]-ps[j]))<mx) {
                mx=abs((ps[j]-ps[x])-(ps[z-1]-ps[j]));
                sum=(ps[j]-ps[x])*(ps[z-1]-ps[j]);
                pos=j;
            }
        }
        u[pos]=true;
        v1.insert(pos),v2.insert(pos+1);
        ansum+=sum;
        ans.push_back(pos);
    }
    cout << ansum << '\n';
    for (auto now : ans)
        cout << now << ' ';
}

signed main() {
    // freopen("sequence.in", "r", stdin);
    // freopen("sequence.out", "w", stdout);
    kanagattandirilmagandiktarinizdan
    int test = 1,count = 1;
    // cin >> test;
    while (test--) {
        // cout << "Case " << count << ": ";
        solve();
        if (test)
            cout << '\n';
        count++;
    }
    return 0;
}
#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...