結果

問題 No.1117 数列分割
ユーザー Plan8Plan8
提出日時 2020-08-02 11:30:04
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 146 ms / 3,000 ms
コード長 2,350 bytes
コンパイル時間 2,023 ms
コンパイル使用メモリ 212,232 KB
実行使用メモリ 78,136 KB
最終ジャッジ日時 2024-07-18 09:47:14
合計ジャッジ時間 4,171 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 11 ms
9,812 KB
testcase_04 AC 10 ms
8,980 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 11 ms
8,308 KB
testcase_09 AC 4 ms
6,944 KB
testcase_10 AC 11 ms
8,300 KB
testcase_11 AC 33 ms
23,112 KB
testcase_12 AC 33 ms
24,808 KB
testcase_13 AC 38 ms
33,188 KB
testcase_14 AC 48 ms
34,136 KB
testcase_15 AC 50 ms
42,012 KB
testcase_16 AC 64 ms
45,912 KB
testcase_17 AC 12 ms
7,736 KB
testcase_18 AC 146 ms
78,136 KB
testcase_19 AC 112 ms
77,896 KB
testcase_20 AC 64 ms
39,296 KB
testcase_21 AC 98 ms
77,888 KB
testcase_22 AC 99 ms
76,760 KB
testcase_23 AC 101 ms
76,800 KB
testcase_24 AC 97 ms
75,692 KB
testcase_25 AC 97 ms
75,520 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 9 ms
6,016 KB
testcase_28 AC 10 ms
5,888 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef vector<int> VI;
typedef vector<VI> VVI;
typedef vector<long long> VL;
typedef vector<vector<long long>> VVL;
typedef vector<string> VS;
typedef pair<ll,ll> P;
typedef tuple<int,int,int> tpl;

#define ALL(a)  (a).begin(),(a).end()
#define SORT(c) sort((c).begin(),(c).end())
#define REVERSE(c) reverse((c).begin(),(c).end())
#define EXIST(m,v) (m).find((v)) != (m).end()
#define LB(a,x) lower_bound((a).begin(), (a).end(), x) - (a).begin()
#define UB(a,x) upper_bound((a).begin(), (a).end(), x) - (a).begin()

#define FOR(i,a,b) for(int i=(a);i<(b);++i)
#define REP(i,n)  FOR(i,0,n)
#define RFOR(i,a,b) for(int i=(a)-1;i>=(b);--i)
#define RREP(i,n) RFOR(i,n,0)

#define en "\n"

constexpr double EPS = 1e-9;
constexpr double PI  = 3.1415926535897932;
constexpr int INF = 2147483647;
constexpr long long LINF = 1LL<<60;
constexpr long long MOD = 1000000007; // 998244353;

template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; }

void Main(){
    ll N,K,M; cin >> N >> K >> M;
    VL A(N); REP(i,N) cin >> A[i];
    VL S(N+1,0); REP(i,N) S[i+1] += S[i]+A[i];
    ll dp[N+1][K+1]; REP(i,N+1)REP(j,K+1) dp[i][j] = -LINF;
    deque<P> deq1[K+1],deq2[K+1];

    dp[0][0] = 0; deq1[0].emplace_back(0,0); deq2[0].emplace_back(0,0);
    REP(i,N)REP(j,min((ll)i+1,K)){
        while(!deq1[j].empty() && deq1[j].front().first<i+1-M) deq1[j].pop_front();
        while(!deq2[j].empty() && deq2[j].front().first<i+1-M) deq2[j].pop_front();

        if(!deq1[j].empty()) chmax(dp[i+1][j+1], deq1[j].front().second + S[i+1]);
        if(!deq2[j].empty()) chmax(dp[i+1][j+1], deq2[j].front().second - S[i+1]);

        while(!deq1[j+1].empty() && deq1[j+1].back().second<=dp[i+1][j+1]-S[i+1]) deq1[j+1].pop_back();
        while(!deq2[j+1].empty() && deq2[j+1].back().second<=dp[i+1][j+1]+S[i+1]) deq2[j+1].pop_back();
        deq1[j+1].emplace_back(i+1,dp[i+1][j+1]-S[i+1]);
        deq2[j+1].emplace_back(i+1,dp[i+1][j+1]+S[i+1]);
    }

    cout << dp[N][K] << en;
    return;
}

int main(void){
    cin.tie(0);cout.tie(0);ios_base::sync_with_stdio(0);cout<<fixed<<setprecision(15);
    int t=1; //cin>>t;
    REP(_,t) Main();
    return 0;
}
0