結果

問題 No.28 末尾最適化
ユーザー koba-e964koba-e964
提出日時 2016-12-26 15:13:41
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 898 ms / 5,000 ms
コード長 2,173 bytes
コンパイル時間 1,189 ms
コンパイル使用メモリ 90,800 KB
実行使用メモリ 4,380 KB
最終ジャッジ日時 2023-08-21 12:42:24
合計ジャッジ時間 2,342 ms
ジャッジサーバーID
(参考情報)
judge14 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,380 KB
testcase_01 AC 898 ms
4,380 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <algorithm>
#include <iostream>
#include <map>
#include <vector>

#define REP(i,s,n) for(int i=(int)(s);i<(int)(n);i++)

using namespace std;
typedef long long int ll;
typedef vector<int> VI;
const ll mod = 1e8 + 9;

void factorize(ll v, std::map<long long, int> &result) {
  long long p = 2;
  while (v > 1 || p * p <= v) {
    int cnt = 0;
    while (v % p == 0) {
      cnt++;
      v /= p;
    }
    if (cnt > 0) {
      if (result.count(p) == 0) {
	result[p] = 0;
      }
      result[p] += cnt;
    }
    p += p == 2 ? 1 : 2;
  }
}

std::pair<std::vector<int>, long long>
factor_base(ll v, const std::vector<long long> &base) {
  int n = base.size();
  std::vector<int> ret(n);
  for (int i = 0; i < n; ++i) {
    ll p = base[i];
    int cnt = 0;
    while (v % p == 0) {
      cnt++;
      v /= p;
    }
    ret[i] = cnt;
  }
  return std::pair<std::vector<int>, long long>(ret, v);
}

// Solved after reading the editorial
int solve(const vector<VI> &tbl, const VI &bias, int k) {
  int n = tbl.size();
  int m = bias.size();
  if (0) {
    REP(i, 0, n) {
      cerr << "(";
      REP(j, 0, m) {
	cerr << " " << tbl[i][j];
      }
      cerr << ")" << endl;
    }
    cerr << "bias =";
    REP(i, 0, m) {
      cerr << " " << bias[i];
    }
    cerr << endl;
  }
  int mi = 1e8;
  REP(i, 0, m) {
    VI tmp(n);
    REP(j, 0, n) {
      tmp[j] = tbl[j][i];
    }
    sort(tmp.begin(), tmp.end());
    int tot = 0;
    REP(j, 0, k) {
      tot += tmp[j];
    }
    mi = min(mi, tot / bias[i]);
  }
  return mi;
}

int main(void){
  int q;
  cin >> q;
  while (q--) {
    ll seed;
    int n, k, b;
    cin >> seed >> n >> k >> b;
    ll cur = seed;
    map<int, int> freq;
    map<ll, int> b_fact;
    factorize(b, b_fact);
    vector<pair<ll, int> > b_fact_v(b_fact.begin(), b_fact.end());
    vector<ll> fbase;
    VI bias;
    REP(i, 0, b_fact_v.size()) {
      fbase.push_back(b_fact_v[i].first);
      bias.push_back(b_fact_v[i].second);
    }
    vector<VI> tbl(n + 1);
    REP(i, 0, n + 1) {
      VI v = factor_base(cur, fbase).first;
      tbl[i] = v;
      cur = 1 + (cur * (cur + 12345) % mod);
    }
    cout << solve(tbl, bias, k) << endl;
  }
}
0