結果

問題 No.2370 He ate many cakes
ユーザー tnakao0123tnakao0123
提出日時 2024-06-26 12:54:22
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,421 bytes
コンパイル時間 1,096 ms
コンパイル使用メモリ 77,780 KB
実行使用メモリ 98,040 KB
最終ジャッジ日時 2024-06-26 12:54:38
合計ジャッジ時間 12,284 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 771 ms
51,808 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 1 ms
6,944 KB
testcase_09 AC 2 ms
6,940 KB
testcase_10 AC 44 ms
9,468 KB
testcase_11 AC 207 ms
23,080 KB
testcase_12 AC 942 ms
50,416 KB
testcase_13 AC 1,598 ms
90,000 KB
testcase_14 AC 380 ms
30,220 KB
testcase_15 TLE -
testcase_16 AC 1,656 ms
87,044 KB
testcase_17 TLE -
testcase_18 AC 267 ms
16,948 KB
testcase_19 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

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

/* constant */

const int MAX_N = 30;
const int NBITS = 1 << MAX_N;
const long long LINF = 1LL << 62;

/* typedef */

using ll = long long;
using pli = pair<ll,int>;
using spli = set<pli>;
using usi = unordered_set<int>;

/* global variables */

int as[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);
  sort(as, as + n);

  int bits0 = 0, m = 0;
  ll sum0 = 0;
  for (int i = 0; i < n; i++) {
    if (as[i] >= 0) {
      bits0 |= (1 << i);
      sum0 += as[i];
    }
    else
      m++;
  }

  spli q;
  q.insert({sum0, bits0});
  usi used;
  used.insert(bits0);
  
  int c = 0;
  ll sumk = 0;
  while (c < k && ! q.empty()) {
    auto sit = q.end(); sit--;
    auto [s, bits] = *sit; q.erase(sit);
    if (++c >= k) { sumk = s; break; }

    for (int i = 0, bi = 1; i < n; i++, bi <<= 1)
      if (used.find(bits ^ bi) == used.end()) {
	ll sum1 = s + ((bits & bi) ? -as[i] : as[i]);
	int bits1 = bits ^ bi;
	used.insert(bits1);
	if (q.size() < k - c)
	  q.insert({sum1, bits1});
	else if (q.begin()->first < sum1) {
	  q.erase(q.begin());
	  q.insert({sum1, bits1});
	}
      }
  }

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