結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 53 ms
191,540 KB
testcase_01 AC 53 ms
191,528 KB
testcase_02 AC 53 ms
191,432 KB
testcase_03 AC 67 ms
200,808 KB
testcase_04 AC 65 ms
200,092 KB
testcase_05 AC 53 ms
191,436 KB
testcase_06 AC 58 ms
191,924 KB
testcase_07 AC 53 ms
191,908 KB
testcase_08 AC 64 ms
200,708 KB
testcase_09 AC 60 ms
195,660 KB
testcase_10 AC 67 ms
201,460 KB
testcase_11 AC 94 ms
218,724 KB
testcase_12 AC 92 ms
219,472 KB
testcase_13 AC 107 ms
226,364 KB
testcase_14 AC 112 ms
230,136 KB
testcase_15 AC 125 ms
235,940 KB
testcase_16 AC 163 ms
243,660 KB
testcase_17 AC 73 ms
207,512 KB
testcase_18 MLE -
testcase_19 MLE -
testcase_20 AC 125 ms
237,384 KB
testcase_21 MLE -
testcase_22 MLE -
testcase_23 MLE -
testcase_24 MLE -
testcase_25 MLE -
testcase_26 AC 54 ms
191,532 KB
testcase_27 WA -
testcase_28 WA -
権限があれば一括ダウンロードができます

ソースコード

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 value;
    Element(int p=0,long long value=0):p(p),value(value){}
};
Element queue[3005][2005][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