結果

問題 No.1117 数列分割
コンテスト
ユーザー hs0319boy
提出日時 2020-07-17 23:42:59
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,175 bytes
コンパイル時間 1,753 ms
コンパイル使用メモリ 175,268 KB
実行使用メモリ 13,644 KB
最終ジャッジ日時 2024-11-30 04:30:36
合計ジャッジ時間 42,741 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15 WA * 1 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll=long long;
using vin=vector<int>;
using vll=vector<long long>;
using vvin=vector<vector<int>>;
using vvll=vector<vector<long long>>;
using vstr=vector<string>;
using vvstr=vector<vector<string>>;
using vch=vector<char>;
using vvch=vector<vector<char>>;
using vbo=vector<bool>;
using vvbo=vector<vector<bool>>;
using vpii=vector<pair<int,int>>;
using pqsin=priority_queue<int,vector<int>,greater<int>>;
#define mp make_pair
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define rep2(i,s,n) for(int i=(s);i<(int)(n);i++)
#define all(v) v.begin(),v.end()
#define decp(n) cout<<fixed<<setprecision((int)n)
const int inf=1e9+7;
const ll INF=1e18;

int main(){
    int n,k,m;cin>>n>>k>>m;
    vll a(n);
    vll num1;
    multiset<ll> num2;
    int tmp=inf;int res;ll sum=(ll)0;int cnt=0;
    rep(i,n){
        cin>>a[i];
        if(a[i]==0)cnt++;
        else if(a[i]<0)res=-1;
        else res=1;
        if(a[i]!=0&&tmp==inf){
            sum+=a[i];
            tmp=res;
            continue;
        }
        if((a[i]!=0&&res!=tmp)||cnt==m){
            num1.push_back(abs(sum));
            num2.insert(abs(sum));
            sum=a[i];
            cnt=1;
            tmp=res;
        }
        else if(a[i]!=0){
            sum+=a[i];
            cnt++;
        }
    }
    num1.push_back(abs(sum));
    num2.insert(abs(sum));
    int s=num1.size();
    while(s>k){
        auto p=*num2.begin();
        rep(i,num1.size()){
            if(num1[i]==p){
                if(i==0||(i!=num1.size()&&num1[i+1]<=num1[i-1])){
                    num2.erase(num2.find(num1[i+1]));
                    num1[i+1]=abs(num1[i+1]-num1[i]);
                    num2.insert(num1[i+1]);
                }
                else{
                    num2.erase(num2.find(num1[i-1]));
                    num1[i-1]=abs(num1[i-1]-num1[i]);
                    num2.insert(num1[i-1]);
                }
                num1.erase(num1.begin()+i);
                num2.erase(num2.find(num1[i]));
                s--;
                break;
            }
        }
    }
    ll ans=(ll)0;
    rep(i,num1.size())ans+=num1[i];
    cout<<ans<<endl;
}
0