Submission #73441

#	Time	Username	Problem	Language	Result	Execution time	Memory
73441		Vardanyan	수열 (APIO14_sequence)	C++14	0 / 100	48 ms	1812 KiB

sequence

//#pragma GCC optimize "-O3"
#include <bits/stdc++.h>
using namespace std;
const int N = 1000*100+7;
int a[N];
long long pref[N];
//long long suf[N];
pair<long long,int> check(int l,int r){
    long long ans = 0;
    int pos = l;
    for(int i = l;i<=r-1;i++){
        long long x = pref[i]-pref[l-1];
        long long y = pref[r]-pref[i];
        x*=y;
        if(x>ans){
            ans = x; pos = i;
        }
    }
    return make_pair(ans,pos);
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i = 1;i<=n;i++) scanf("%d",&a[i]);
    for(int i = 1;i<=n;i++) pref[i] = pref[i-1]+a[i];
//    for(int i = n;i>=1;i--) suf[i] = suf[i+1]+a[i];
    vector<pair<int,int> > v;
    v.push_back({1,n});
    long long ans = 0;
    vector<int> p;
    while(v.size()<=k){
        pair<long long,int> mxans;
        for(int i = 0;i<v.size();i++){
            int l = v[i].first;
            int r = v[i].second;
            mxans = max(mxans,check(l,r));
        }
        ans+=mxans.first;
        p.push_back(mxans.second);
        for(int i = 0;i<v.size();i++){
            int l = v[i].first;
            int r = v[i].second;
            if(mxans.second>=l && mxans.second<=r){
                v.erase(v.begin()+i);
                v.push_back(make_pair(l,mxans.second));
                v.push_back(make_pair(mxans.second+1,r));
                break;
            }
        }
    }
    printf("%lld\n",ans);
    for(int i = 0;i<p.size();i++) printf("%d ",p[i]);
    printf("\n");
    return 0;
}
/*
7 3
4 1 3 4 0 2 3
*/

Compilation message (stderr)

sequence.cpp: In function 'int main()':
sequence.cpp:32:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     while(v.size()<=k){
           ~~~~~~~~^~~
sequence.cpp:34:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i = 0;i<v.size();i++){
                       ~^~~~~~~~~
sequence.cpp:41:24: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
         for(int i = 0;i<v.size();i++){
                       ~^~~~~~~~~
sequence.cpp:53:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i = 0;i<p.size();i++) printf("%d ",p[i]);
                   ~^~~~~~~~~
sequence.cpp:24:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d",&n,&k);
     ~~~~~^~~~~~~~~~~~~~
sequence.cpp:25:34: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     for(int i = 1;i<=n;i++) scanf("%d",&a[i]);
                             ~~~~~^~~~~~~~~~~~

• Subtask 1: 0 / 11
• Subtask 2: 0 / 11
• Subtask 3: 0 / 11
• Subtask 4: 0 / 17
• Subtask 5: 0 / 21
• Subtask 6: 0 / 29