結果

問題 No.1463 Hungry Kanten
ユーザー tnakao0123tnakao0123
提出日時 2021-04-04 02:22:20
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 811 ms / 2,000 ms
コード長 2,811 bytes
コンパイル時間 1,750 ms
コンパイル使用メモリ 111,380 KB
実行使用メモリ 55,572 KB
最終ジャッジ日時 2024-06-07 17:25:31
合計ジャッジ時間 4,053 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 23 ms
5,376 KB
testcase_03 AC 301 ms
31,984 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 811 ms
55,572 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 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 3 ms
5,376 KB
testcase_13 AC 2 ms
5,376 KB
testcase_14 AC 8 ms
5,376 KB
testcase_15 AC 2 ms
5,376 KB
testcase_16 AC 24 ms
5,376 KB
testcase_17 AC 70 ms
8,688 KB
testcase_18 AC 22 ms
5,376 KB
testcase_19 AC 342 ms
25,472 KB
testcase_20 AC 483 ms
33,380 KB
testcase_21 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 1463.cc:  No.1463 Hungry Kanten - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 18;
const int MAX_A = 1000;
const int MAX_P = MAX_N * MAX_A;

/* typedef */

typedef pair<int,int> pii;
typedef vector<pii> vpii;
typedef vector<vpii> vvpii;

/* global variables */

bool primes[MAX_P + 1];
int pnums[MAX_P], as[MAX_N];
vpii apds[MAX_N];

/* subroutines */

int gen_primes(int maxp, int pnums[]) {
  memset(primes, true, sizeof(primes));
  primes[0] = primes[1] = false;

  int p, pn = 0;
  for (p = 2; p * p <= maxp; p++)
    if (primes[p]) {
      pnums[pn++] = p;
      for (int q = p * p; q <= maxp; q += p) primes[q] = false;
    }
  for (; p <= maxp; p++)
    if (primes[p]) pnums[pn++] = p;
  return pn;
}

bool prime_decomp(int n, int pn, int pnums[], vpii& pds) {
  pds.clear();

  for (int i = 0; i < pn; i++) {
    int pi = pnums[i];
    if (pi * pi > n) {
      if (n > 1) pds.push_back(pii(n, 1));
      return true;
    }

    if (n % pi == 0) {
      int fi = 0;
      while (n % pi == 0) n /= pi, fi++;
      pds.push_back(pii(pi, fi));
    }
  }
  return false;
}

int bnum(int n) {
  int c = 0;
  for (; n > 0; n >>= 1) c += (n & 1);
  return c;
}

void mulpd(vpii &a, vpii &b, vpii &c) {
  c.clear();
  int an = a.size(), bn = b.size();
  int ai = 0, bi = 0;
  while (ai < an && bi < bn) {
    if (a[ai].first < b[bi].first) c.push_back(a[ai++]);
    else if (a[ai].first > b[bi].first) c.push_back(b[bi++]);
    else {
      c.push_back(pii(a[ai].first, a[ai].second + b[bi].second));
      ai++, bi++;
    }
  }

  while (ai < an) c.push_back(a[ai++]);
  while (bi < bn) c.push_back(b[bi++]);
}

/* main */

int main() {
  int pn = gen_primes(MAX_P, pnums);
  //printf("pn=%d\n", pn);

  int n, k;
  scanf("%d%d", &n, &k);

  for (int i = 0; i < n; i++) {
    scanf("%d", as + i);
    prime_decomp(as[i], pn, pnums, apds[i]);
  }

  int nbits = 1 << n;
  vvpii ans;

  for (int bits = 0; bits < nbits; bits++) {
    if (bnum(bits) < k) continue;

    int sum = 0;
    vpii ppd;
    for (int i = 0, bi = 1; i < n; i++, bi <<= 1)
      if (bits & bi) {
	sum += as[i];
	if (ppd.empty()) ppd = apds[i];
	else {
	  vpii ppd0;
	  mulpd(ppd, apds[i], ppd0);
	  swap(ppd, ppd0);
	}
      }

    vpii spd;
    prime_decomp(sum, pn, pnums, spd);

    ans.push_back(spd);
    ans.push_back(ppd);
  }

  sort(ans.begin(), ans.end());
  ans.erase(unique(ans.begin(), ans.end()), ans.end());

  printf("%d\n", (int)ans.size());
  return 0;
}
0