結果

問題 No.2028 Even Choice
コンテスト
ユーザー tnakao0123
提出日時 2022-08-13 01:53:18
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 130 ms / 2,000 ms
コード長 1,165 bytes
コンパイル時間 507 ms
コンパイル使用メモリ 58,496 KB
実行使用メモリ 14,976 KB
最終ジャッジ日時 2024-09-23 06:00:43
合計ジャッジ時間 3,657 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2028.cc:  No.2028 Even Choice - yukicoder
 */

#include<cstdio>
#include<set>
#include<algorithm>
#include<utility>
 
using namespace std;

/* constant */

const int MAX_N = 200000;

/* typedef */

typedef long long ll;
typedef pair<int,int> pii;
typedef set<pii> spii;

/* global variables */

int as[MAX_N];
pii ps[MAX_N];

/* subroutines */

/* main */

int main() {
  int n, k;
  scanf("%d%d", &n, &k);
  for (int i = 0; i < n; i++) scanf("%d", as + i);

  for (int i = 0; i < n; i++) ps[i] = pii(as[i], i);
  sort(ps, ps + n, greater<pii>());

  ll sum = 0;
  spii q;
  for (int i = 0; i < k - 1; i++) {
    sum += ps[i].first;
    q.insert(pii(ps[i].second, ps[i].first));
  }

  ll maxs = 0;
  for (int i = 1, j = k - 1; i < n; i += 2) {
    while (! q.empty() && q.begin()->first <= i) {
      sum -= q.begin()->second;
      q.erase(q.begin());
    }

    while (j < n && q.size() < k - 1) {
      if (ps[j].second > i) {
	sum += ps[j].first;
	q.insert(pii(ps[j].second, ps[j].first));
      }
      j++;
    }
    if (q.size() < k - 1) break;

    maxs = max(maxs, sum + as[i]);
  }

  printf("%lld\n", maxs);
  return 0;
}
0