結果

問題 No.2370 He ate many cakes
ユーザー tnakao0123tnakao0123
提出日時 2024-06-26 10:25:35
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 1,288 bytes
コンパイル時間 650 ms
コンパイル使用メモリ 75,252 KB
実行使用メモリ 98,040 KB
最終ジャッジ日時 2024-06-26 10:25:53
合計ジャッジ時間 15,311 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,376 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 1,174 ms
51,808 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 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 67 ms
9,476 KB
testcase_11 AC 356 ms
23,104 KB
testcase_12 AC 1,256 ms
50,416 KB
testcase_13 TLE -
testcase_14 AC 615 ms
30,220 KB
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 AC 271 ms
16,948 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<set>
#include<unordered_set>
#include<algorithm>
#include<utility>
 
using namespace std;

/* constant */

const int MAX_N = 30;
const int NBITS = 1 << MAX_N;

/* 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);

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

  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;
	q.insert({sum1, bits1});
	used.insert(bits1);
      }
    
    while (q.size() > k - c) q.erase(q.begin());
  }

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