結果

問題 No.1117 数列分割
ユーザー DriceDrice
提出日時 2020-07-18 03:22:30
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 2,198 bytes
コンパイル時間 459 ms
コンパイル使用メモリ 34,816 KB
実行使用メモリ 356,000 KB
最終ジャッジ日時 2024-05-07 22:46:41
合計ジャッジ時間 5,884 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

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

ソースコード

diff #

#include<cstdio>
//abs part can be fixed...
long long a[3005];
long long sum[3005];
long long f[3005][3005];
struct Element{
    int p;
    long long value;
    Element(int p=0,long long value=0):p(p),value(value){}
};
Element queue[3005][3005][2];
int h[3005][2], t[3005][2];

long long abs(long long u){ return u>0?u:-u; }

long long max(long long a,long long b){ return a>b?a:b; }

int main(){
    int n,m,k;
    scanf("%d%d%d",&n,&k,&m);
    sum[0] = 0;
    for(int i = 1; i <= n; i++){
        scanf("%lld",&a[i]);
        sum[i] = sum[i-1]+a[i];
    }
    long long inf = 1e15+7;
    for(int i = 0; i <= k; i++){ 
        f[0][i] = -inf;
        h[i][0] = h[i][1] = 0;
        t[i][0] = t[i][1] = -1;
    }
    f[0][0] = 0;
    queue[0][++t[0][0]][0] = Element(0,0);
    queue[0][++t[0][1]][1] = Element(0,0);
    for(int i = 1; i <= n; i++){
        f[i][0] = -inf;
        for(int j = 1; j <= k; j++){
            f[i][j] = -inf;
            /*for(int last = max(0,i-m); last <= i-1; last++){
                //f[i][j] = max(f[i][j],f[last][j-1]+abs(sum[i]-sum[last]));
                //f[i][j] = max(f[i][j],f[last][j-1]+sum[i]-sum[last]);
                //f[i][j] = max(f[i][j],f[last][j-1]-sum[i]+sum[last]);
            }*/
            while(h[j-1][0]<=t[j-1][0] && queue[j-1][h[j-1][0]][0].p<i-m) h[j-1][0]++;
            while(h[j-1][1]<=t[j-1][1] && queue[j-1][h[j-1][1]][1].p<i-m) h[j-1][1]++;
            long long res0 = -inf, res1 = -inf;
            if(h[j-1][0]<=t[j-1][0]) res0 = queue[j-1][h[j-1][0]][0].value+sum[i];
            if(h[j-1][1]<=t[j-1][1]) res1 = queue[j-1][h[j-1][1]][1].value-sum[i];
            f[i][j] = max(res0,res1);
            //printf("f[%d][%d] = %lld\n",i,j,f[i][j]);
            //update
        }
        for(int j = 1; j <= k; j++){
            long long value = f[i][j]-sum[i];
            while(t[j][0]>=0 && value>=queue[j][t[j][0]][0].value) t[j][0]--;
            queue[j][++t[j][0]][0] = Element(i,value);
            value = f[i][j]+sum[i];
            while(t[j][1]>=0 && value>=queue[j][t[j][1]][1].value) t[j][1]--;
            queue[j][++t[j][1]][1] = Element(i,value);
        }
    }
    printf("%lld\n",f[n][k]);
    return 0;
}
0