結果

問題 No.2370 He ate many cakes
ユーザー tnakao0123tnakao0123
提出日時 2024-06-26 13:17:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 46 ms / 2,000 ms
コード長 1,390 bytes
コンパイル時間 846 ms
コンパイル使用メモリ 77,352 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-26 13:17:39
合計ジャッジ時間 1,881 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 20 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 1 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 7 ms
5,376 KB
testcase_11 AC 13 ms
5,376 KB
testcase_12 AC 23 ms
5,376 KB
testcase_13 AC 34 ms
5,376 KB
testcase_14 AC 14 ms
5,376 KB
testcase_15 AC 44 ms
5,376 KB
testcase_16 AC 46 ms
5,376 KB
testcase_17 AC 45 ms
5,376 KB
testcase_18 AC 23 ms
5,376 KB
testcase_19 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2370.cc:  No.2370 He ate many cakes - yukicoder
 */

#include<cstdio>
#include<queue>
#include<algorithm>
#include<utility>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 30;
const int MAX_M = MAX_N / 2;
const int MBITS = 1 << MAX_M;

/* typedef */

using ll = long long;
using pli = pair<ll,int>;

/* global variables */

int as[MAX_N];
ll s0[MBITS], s1[MBITS];

/* subroutines */

/* main */

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

  int m0 = (n + 1) / 2, m1 = n - m0;
  int mbits0 = 1 << m0, mbits1 = 1 << m1;

  s0[0] = 0;
  for (int bits = 1, msb = 1, msi = 0; bits < mbits0; bits++) {
    if ((msb << 1) <= bits) msb <<= 1, msi++;
    s0[bits] = s0[bits ^ msb] + as[msi];
  }
  sort(s0, s0 + mbits0, greater<ll>());

  s1[0] = 0;
  for (int bits = 1, msb = 1, msi = 0; bits < mbits1; bits++) {
    if ((msb << 1) <= bits) msb <<= 1, msi++;
    s1[bits] = s1[bits ^ msb] + as[m0 + msi];
  }
  sort(s1, s1 + mbits1, greater<ll>());

  priority_queue<pli> q;
  for (int i = 0; i < mbits0; i++)
    q.push({s0[i] + s1[0], 0});

  int c = 0;
  ll sk = 0;
  while (c < k) {
    auto [s, u] = q.top(); q.pop();
    if (++c >= k) { sk = s; break; }

    if (u + 1 < mbits1)
      q.push({s - s1[u] + s1[u + 1], u + 1});
  }

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