結果

問題 No.1731 Product of Subsequence
ユーザー hiro71687k
提出日時 2023-03-31 00:44:40
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,127 ms / 2,000 ms
コード長 1,239 bytes
コンパイル時間 4,181 ms
コンパイル使用メモリ 259,080 KB
最終ジャッジ日時 2025-02-11 19:25:47
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using ld=long double;
ld pie=3.141592653589793;
ll mod=1000000007;
ll inf=999999999999999999;
ll gcd(ll a, ll b) {
	a = abs(a); b = abs(b);
	if (a < b)swap(a, b);
	while (b) {
	    ll r=a%b;
        a=b;
	    b=r;
	}
	return a;
}
ll modpow(ll x, ll n) {
  x = x%mod;
  if(n==0) return 1;  //再帰の終了条件

  else if(n%2==1) {
    return (x*modpow(x, n-1))%mod;  //nが奇数ならnを1ずらす
  }
  else return modpow((x*x)%mod, n/2)%mod;  //nが偶数ならnが半分になる
}
int main(){
    ll n,k;
    cin >> n >> k;
    if (k==1)
    {
        cout << (modpow(2,n)+mod-1)%mod << endl;
        return 0;
    }
    
    vector<ll>a(n);
    for (ll i = 0; i < n; i++)
    {
        cin >> a[i];
        a[i]=gcd(k,a[i]);
    }
    vector<map<ll,ll>>dp(n);
    dp[0][1]=1;
    ll x=gcd(k,a[0]);
    dp[0][x]+=1;
    for (ll i = 1; i < n; i++)
    {
        for(auto v:dp[i-1]){
            dp[i][v.first]+=v.second;
            dp[i][v.first]%=mod;
            ll z=v.first*a[i];
            z=gcd(k,z);
            dp[i][z]+=v.second;
            dp[i][z]%=mod;
        }
    }
    cout << dp[n-1][k] << endl;
}
0