제출 #114180

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
114180		patrikpavic2	수열 (APIO14_sequence)	C++17	71 / 100	2058 ms	85240 KiB

sequence

#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <algorithm>
#include <algorithm>
#include <deque>

#define X first
#define Y second
#define PB push_back

using namespace std;

typedef long long ll;
typedef pair < ll, ll > pll;

const int N = 1e5 + 500;
const int K = 205;

inline bool ccw(const pll &A, const pll &B, const pll &C){
    return A.X * (B.Y - C.Y) + B.X * (C.Y - A.Y) + C.X * (A.Y - B.Y) <= 0LL;
}

int ind[N];
pll hl[N];
int cur = 0, n, k, a[N];
ll P[N], lst[N], dp[N];
int rek[N][K], cur_k, sz;

inline void add(const pll &X,const int &i){
    for(;sz >= 2 && ccw(X, hl[sz - 1], hl[sz - 2]); sz--);
    hl[sz] = X; ind[sz++] = i;
    cur = min(cur, sz - 1);
}

inline void dinamika(){
    sz = 0; cur = 0;
    for(int i = 0;i < n;i++){
        dp[i] = -1;
        if(sz > 0){
            for(; cur + 1 < sz && hl[cur].X * P[i] + hl[cur].Y < hl[cur + 1].X * P[i] + hl[cur + 1].Y; cur++);
            rek[i][cur_k] = ind[cur];
            dp[i] = hl[cur].X * P[i] + hl[cur].Y;
        }
        if(lst[i] >= 0)
            add({P[i], lst[i] - P[i] * P[i]}, i);
        lst[i] = dp[i];
    }
    cur_k++;
}

int main(){
    scanf("%d%d", &n, &k);
    for(int i = 0;i < n;i++){
        scanf("%d", a + i);
        P[i] = a[i] + (i ? P[i - 1] : 0);
    }
    for(int i = 0;i < k;i++)
        dinamika();
    printf("%lld\n", dp[n - 1]);
    vector < int > fin;
    int cur = n - 1;
    for(int i = k - 1;i >= 0;i--){
        fin.PB(rek[cur][i]);
        cur = fin.back();
    }
    reverse(fin.begin(), fin.end());
    for(int x : fin)
        printf("%d ", x + 1);
    printf("\n");
}

컴파일 시 표준 에러 (stderr) 메시지

sequence.cpp: In function 'int main()':
sequence.cpp:58:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d", &n, &k);
     ~~~~~^~~~~~~~~~~~~~~~
sequence.cpp:60:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d", a + i);
         ~~~~~^~~~~~~~~~~~~

• Subtask 1: 11 / 11
• Subtask 2: 11 / 11
• Subtask 3: 11 / 11
• Subtask 4: 17 / 17
• Subtask 5: 21 / 21
• Subtask 6: 0 / 29