結果

問題 No.1731 Product of Subsequence
ユーザー eQeeQe
提出日時 2021-11-06 22:13:33
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 593 ms / 2,000 ms
コード長 3,298 bytes
コンパイル時間 4,733 ms
コンパイル使用メモリ 239,856 KB
実行使用メモリ 14,080 KB
最終ジャッジ日時 2024-11-08 05:03:21
合計ジャッジ時間 12,643 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 592 ms
13,696 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 3 ms
5,248 KB
testcase_08 AC 31 ms
5,248 KB
testcase_09 AC 3 ms
5,248 KB
testcase_10 AC 579 ms
14,080 KB
testcase_11 AC 496 ms
12,544 KB
testcase_12 AC 4 ms
5,248 KB
testcase_13 AC 6 ms
5,248 KB
testcase_14 AC 583 ms
11,520 KB
testcase_15 AC 2 ms
5,248 KB
testcase_16 AC 3 ms
5,248 KB
testcase_17 AC 2 ms
5,248 KB
testcase_18 AC 581 ms
14,080 KB
testcase_19 AC 7 ms
5,248 KB
testcase_20 AC 468 ms
12,032 KB
testcase_21 AC 575 ms
14,080 KB
testcase_22 AC 593 ms
14,080 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 17 ms
5,248 KB
testcase_25 AC 9 ms
5,248 KB
testcase_26 AC 101 ms
5,632 KB
testcase_27 AC 148 ms
6,144 KB
testcase_28 AC 303 ms
8,576 KB
testcase_29 AC 107 ms
5,760 KB
testcase_30 AC 581 ms
13,952 KB
testcase_31 AC 585 ms
13,952 KB
testcase_32 AC 3 ms
5,248 KB
testcase_33 AC 26 ms
5,248 KB
testcase_34 AC 25 ms
5,248 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'int main()':
main.cpp:62:14: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17' [-Wc++17-extensions]
   62 |     for(auto [p,_]:prim){
      |              ^

ソースコード

diff # 

#include <atcoder/all>
using namespace atcoder;
#include <bits/stdc++.h>
#define all(a) (a.begin()),(a.end())
#define rall(a) (a.rbegin()),(a.rend())
using namespace std;
using ll=long long;
using ld=long double;
using pll=pair<ll,ll>;
template<class T>bool chmax(T &a, const T &b) {if(a<b) {a=b; return 1;} return 0;}
template<class T>bool chmin(T &a, const T &b) {if(b<a) {a=b; return 1;} return 0;}
template<class T1, class T2>istream& operator >> (istream& is, pair<T1,T2>& p) { is >> p.first >> p.second; return is; }
template<class T1, class T2>ostream& operator << (ostream& os, const pair<T1,T2>& p) { os << p.first << " " << p.second; return os;}
template<class T>istream& operator >> (istream& is, vector<T>& v) { for(T& x:v) is >> x; return is; }
template<class T>ostream& operator << (ostream& os, const vector<T>& v) {for(int i=0;i<v.size();i++) {os << v[i] << (i+1 == v.size() ? "":" ");} return os;}

//istream& operator >> (istream& is, modint1000000007& x) { unsigned int t; is >> t; x=t; return is; }
//istream& operator >> (istream& is, modint998244353& x) { unsigned int t; is >> t; x=t; return is; }
ostream& operator << (ostream& os, const modint1000000007& x) { os << x.val(); return os; }
ostream& operator << (ostream& os, const modint998244353& x) { os << x.val(); return os; }
template<class... A> void pt()      { std::cout << "\n"; }
template<class... A> void pt_rest() { std::cout << "\n"; }
template<class T, class... A> void pt_rest(const T& first, const A&... rest) { std::cout << " " << first; pt_rest(rest...); }
template<class T, class... A> void pt(const T& first, const A&... rest)      { std::cout << first; pt_rest(rest...); }

static const ll INF=1e18+7;
static const double PI=acos(-1);
static const double EPS=1e-10;

using v1d=vector<ll>;
using v2d=vector<v1d>;
using v3d=vector<v2d>;
using v4d=vector<v3d>;

vector<pll>prime_factrization(ll n){
  vector<pll>res;
  for(ll i=2;i*i<=n;i++){
    ll cnt=0;
    while(n%i==0){
      cnt++;
      n/=i;
    }
    if(cnt)res.push_back({i,cnt});
  }
  if(n>1)res.push_back({n,1});
  return res;
}

int main(void) {
  cin.tie(nullptr);
  ios::sync_with_stdio(false);
    
  ll N,K;cin>>N>>K;
  v1d a(N);cin>>a;
  
  vector<pll>prim=prime_factrization(K);
  ll M=prim.size();
  
  v2d cnt(N);
  for(ll i=0;i<N;i++){
    v1d v;
    for(auto [p,_]:prim){
      ll t=0;
      while(a[i]%p==0)t++,a[i]/=p;
      v.push_back(t);
    }
    cnt[i]=v;
  }
  
  auto encode=[&](v1d v){
    ll res=0;
    for(ll i=0;i<M-1;i++){
      ll base=prim[i+1].second+1;
      res+=v[i];
      res*=base;
    }
    res+=v.back();
    return res;
  };
  
  auto decode=[&](ll S){
    v1d res;
    for(ll i=M-1;i>=1;i--){
      ll base=prim[i].second+1;
      res.push_back(S%base);
      S/=base;
    }
    res.push_back(S);
    reverse(all(res));
    return res;
  };
  
  ll S=1;
  for(ll i=0;i<M;i++)S*=prim[i].second+1;
  using mint=modint1000000007;
  vector<vector<mint>>dp(N+1,vector<mint>(S));
  dp[0][0]=1;
  for(ll i=0;i<N;i++){
    for(ll j=0;j<S;j++){
      dp[i+1][j]+=dp[i][j];
      
      v1d v=decode(j);
      for(ll k=0;k<M;k++){
        v[k]+=cnt[i][k];
        chmin(v[k],prim[k].second);
      }
      
      ll nxt=encode(v);
      if(nxt<S)dp[i+1][nxt]+=dp[i][j];
    }
  }
  
  pt(dp[N][S-1]-(K==1));
  
}
0