結果

問題 No.1463 Hungry Kanten
ユーザー noya2noya2
提出日時 2021-04-02 21:47:34
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,018 ms / 2,000 ms
コード長 4,135 bytes
コンパイル時間 5,086 ms
コンパイル使用メモリ 280,184 KB
実行使用メモリ 196,224 KB
最終ジャッジ日時 2024-06-06 09:18:58
合計ジャッジ時間 8,605 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 33 ms
9,088 KB
testcase_03 AC 451 ms
5,376 KB
testcase_04 AC 4 ms
5,376 KB
testcase_05 AC 1,018 ms
196,224 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 3 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 3 ms
5,376 KB
testcase_12 AC 4 ms
5,376 KB
testcase_13 AC 5 ms
5,376 KB
testcase_14 AC 10 ms
5,376 KB
testcase_15 AC 11 ms
5,376 KB
testcase_16 AC 34 ms
10,880 KB
testcase_17 AC 84 ms
21,760 KB
testcase_18 AC 93 ms
8,448 KB
testcase_19 AC 415 ms
97,664 KB
testcase_20 AC 680 ms
117,760 KB
testcase_21 AC 2 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
/*
#include <boost/multiprecision/cpp_dec_float.hpp>
#include <boost/multiprecision/cpp_int.hpp>
namespace mp = boost::multiprecision;
using bint = mp::cpp_int;
*/
#include <atcoder/all>
#include <iostream>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <set>
#include <map>
#include <random>
#define rep(i,n) for (int i = 0; i < (n); ++i)
#define repp(i,n,m) for (int i = m; i < (n); ++i)
using namespace std;
using namespace atcoder;
using namespace internal;
//alias g++='g++ -I/mnt/c/Users/Owner/Desktop/ac-library'
using ll = long long;
using ld = long double;
using P = pair<int, int>;
using PI = pair<pair<int,int>,int>;
using PL = pair<long long, long long>;
using PLL = pair<pair<long long, long long>, long long>;
using Pxy = pair<long double, long double>;
const int INF = 1001001007;
const int modd = 1000000007;
const long long modl = 1000000007LL;
const long long mod = 998244353LL;
const ll inf = 2e18;

template <typename SA>
void priv(vector<SA> &ar){
    rep(i,ar.size()-1) cout << ar[i] << " ";
    cout << ar[ar.size()-1] << endl;
}
template <typename SB>
void privv(vector<vector<SB>> &ar){
    rep(i,ar.size()){
        rep(j,ar[i].size()-1) cout << ar[i][j] << " ";
        cout << ar[i][ar[i].size()-1] << endl;
    }
}
template <typename SC>
bool range(SC a, SC b, SC x){return (a <= x && x < b);}
bool rrange(P a, P b, P xy){
    bool s = range(a.first,b.first,xy.first);
    bool t = range(a.second,b.second,xy.second);
    return (s && t);
}
template <typename SD>
void sor(vector<SD> &ar){sort(ar.begin(),ar.end());}
template <typename SE>
void rev(vector<SE> &ar){reverse(ar.begin(),ar.end());}
template <typename SF>
bool chmin(SF &a, const SF &b){if(a>b){a = b; return true;} return false;}
template <typename SG>
bool chmax(SG &a, const SG &b){if(a<b){a = b; return true;} return false;}
template <typename SH>
void eru(vector<SH> &ar){sor(ar);ar.erase(unique(ar.begin(),ar.end()),ar.end());}
template <typename SI>
SI   last(vector<SI> &ar){return ar[ar.size()-1];}
template <typename SJ>
SJ   cel(SJ a, SJ b){if (a % b == 0) return a/b; return a/b +1;}
template <typename SK, typename SL>
void pout(pair<SK,SL> p) {cout << p.first << " " << p.second << endl;}

void yes(){cout << "Yes" << endl;}
void no (){cout << "No" << endl;}
void yn (bool t){if(t)yes();else no();}
void Yes(){cout << "YES" << endl;}
void No (){cout << "NO" << endl;}
void YN (bool t){if(t)Yes();else No();}
void dout() {cout << setprecision(20);}

vector<int> dx = {0,1,0,-1};
vector<int> dy = {1,0,-1,0};
const string ALP = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";
const string alp = "abcdefghijklmnopqrstuvwxyz";

ll gcds(ll a, ll b){
    ll c = a % b;
    while (c != 0){
        a = b;
        b = c;
        c = a % b;
    }
    return b;
}

int main(){
    int n, k; cin >> n >> k;
    vector<int> ar(n);
    rep(i,n) cin >> ar[i];
    set<int> st;
    rep(i,1<<n){
        int cnt = 0;
        int wa = 0;
        rep(j,n){
            if (i >> j & 1){
                cnt++;
                wa += ar[j];
            }
        }
        if (cnt >= k) st.insert(wa);
    }
    vector<int> pri;
    repp(i,1001,1) if (is_prime_constexpr(i)) pri.emplace_back(i);
    map<int,int> mp;
    int m = pri.size();
    rep(i,m) mp[pri[i]] = i;
    vector<vector<int>> soi(n,vector<int>(m,0));
    rep(i,n){
        int ima = ar[i];
        rep(j,m){
            while (ima % pri[j] == 0){
                ima /= pri[j];
                soi[i][j]++;
            }
            if (ima == 1) break;
        }
    }
    set<vector<int>> stv;
    rep(i,1<<n){
        int cnt = 0;
        int seki = 1;
        bool ok = true;
        vector<int> ima(m,0);
        rep(j,n){
            if (i >> j & 1){
                cnt++;
                rep(l,m) ima[l] += soi[j][l];
                if (ok) seki *= ar[j];
                if (seki >= 20000) ok = false;
            }
        }
        if (cnt >= k){
            if (ok) st.insert(seki);
            else {
                stv.insert(ima);
            }
        }
    }
    int ans = st.size() + stv.size();
    cout << ans << endl;
}
0