//#pragma GCC target("avx2") //#pragma GCC optimize("O3") //#pragma GCC optimize("unroll-loops") #include using namespace std; using ll = long long; using pii = pair; using pll = pair; using pli = pair; using vi = vector; using vl = vector; #define TEST cerr << "TEST" << endl #define AMARI 998244353 //#define AMARI 1000000007 #define el '\n' #define El '\n' #define YESNO(x) ((x) ? "Yes" : "No") #define YES YESNO(true) #define NO YESNO(false) #define VEC_UNIQ(v) {sort(v.begin(),v.end()); v.erase(unique(v.begin(),v.end()),v.end());} #define REV_PRIORITY_QUEUE(tp) priority_queue,greater> #define NO_ANS(x) {cout << (x) << '\n'; return;} #define MULTI_TEST_CASE false void solve(void){ //問題を見たらまず「この問題設定から言えること」をいっぱい言う //一個回答に繋がりそうな解法が見えても、実装や細かい詰めに時間がかかりそうなら別の方針を考えてみる //添え字回りで面倒になりそうなときは楽になる言い換えを実装の前にじっくり考える //ある程度考察しても全然取っ掛かりが見えないときは実験をしてみる //よりシンプルな問題に言い換えられたら、言い換えた先の問題を自然言語ではっきりと書く //g++ -D_GLIBCXX_DEBUG -O2 c.cpp -o o int n; cin >> n; ll k; cin >> k; //abs(a_i) <= 1e7 vector a(n); for(int i = 0; i < n; i++)cin >> a[i]; //1 個ずつ取る場合に比べて、(N - K - 1)個余計に取れる (K == N-1) の時0個 deque deq; ll deq_sum = 0LL; for(int i = 0; i < n - k; i++){ deq.push_back(a[i]); deq_sum += a[i]; } int idx = n - k; ll ans = 0LL; //添え字を合わせたいので for(ll i = 1; i <= k; i++){ //cerr << deq_sum << ' ' << ans << el; bool popped = false; while((int)deq.size() >= 1 && deq_sum <= 0){ ans += i * deq.front(); deq_sum -= deq.front(); deq.pop_front(); popped = true; } if(!popped){ ans += i * deq.front(); deq_sum -= deq.front(); deq.pop_front(); } if(idx != n){ deq.push_back(a[idx]); deq_sum += a[idx]; idx++; } } while(!deq.empty() && deq_sum >= 0){ ans += deq.front() * k; deq.pop_front(); } cout << ans << el; return; } void calc(void){ return; } signed main(void){ cin.tie(nullptr); ios::sync_with_stdio(false); calc(); int t = 1; if(MULTI_TEST_CASE)cin >> t; while(t--){ solve(); } return 0; }