結果

問題 No.1117 数列分割
ユーザー hs0319boyhs0319boy
提出日時 2020-07-17 23:44:33
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,563 bytes
コンパイル時間 1,950 ms
コンパイル使用メモリ 176,328 KB
実行使用メモリ 10,752 KB
最終ジャッジ日時 2024-05-07 11:59:04
合計ジャッジ時間 6,558 ms
ジャッジサーバーID
(参考情報)
judge1 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,752 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 TLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
権限があれば一括ダウンロードができます

ソースコード

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){
                    num2.erase(num2.find(num1[i+1]));
                    num1[i+1]=abs(num1[i+1]-num1[i]);
                    num2.insert(num1[i+1]);
                }
                else if(i==num1.size()){
                    num2.erase(num2.find(num1[i-1]));
                    num1[i-1]=abs(num1[i-1]-num1[i]);
                    num2.insert(num1[i-1]);
                }
                else if(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