結果

問題 No.1117 数列分割
ユーザー DriceDrice
提出日時 2020-07-18 03:28:04
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
MLE  
実行時間 -
コード長 2,193 bytes
コンパイル時間 212 ms
コンパイル使用メモリ 35,072 KB
実行使用メモリ 262,184 KB
最終ジャッジ日時 2024-11-30 19:46:06
合計ジャッジ時間 6,545 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 89 ms
191,468 KB
testcase_01 AC 91 ms
191,476 KB
testcase_02 AC 88 ms
191,484 KB
testcase_03 AC 104 ms
200,784 KB
testcase_04 AC 100 ms
200,064 KB
testcase_05 AC 88 ms
191,352 KB
testcase_06 AC 88 ms
191,872 KB
testcase_07 AC 88 ms
192,020 KB
testcase_08 AC 101 ms
200,472 KB
testcase_09 AC 92 ms
195,756 KB
testcase_10 AC 102 ms
201,256 KB
testcase_11 AC 137 ms
218,624 KB
testcase_12 AC 139 ms
219,392 KB
testcase_13 AC 176 ms
226,364 KB
testcase_14 AC 171 ms
230,244 KB
testcase_15 AC 211 ms
235,992 KB
testcase_16 AC 215 ms
243,692 KB
testcase_17 AC 106 ms
207,488 KB
testcase_18 MLE -
testcase_19 MLE -
testcase_20 AC 263 ms
237,432 KB
testcase_21 MLE -
testcase_22 MLE -
testcase_23 MLE -
testcase_24 MLE -
testcase_25 MLE -
testcase_26 AC 89 ms
191,488 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