結果

問題 No.1929 Exponential Sequence
ユーザー asaringo
提出日時 2022-05-07 02:03:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 1,627 bytes
コンパイル時間 2,417 ms
コンパイル使用メモリ 202,016 KB
最終ジャッジ日時 2025-01-29 04:22:52
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std ;
#define fast_input_output ios::sync_with_stdio(false); cin.tie(nullptr);
typedef long long ll ;
typedef long double ld ;
typedef pair<ll,ll> P ;
typedef tuple<ll,ll,ll> TP ;
#define chmin(a,b) a = min(a,b)
#define chmax(a,b) a = max(a,b)
#define bit_count(x) __builtin_popcountll(x)
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a / gcd(a,b) * b
#define rep(i,n) for(int i = 0 ; i < n ; i++)
#define rrep(i,a,b) for(int i = a ; i < b ; i++)
#define endl "\n"

int n ;
ll s ;
ll A[10] ;
map<ll,ll> dp[5] , ep[5] ;
vector<ll> X , Y ;

int main(){
    fast_input_output
    cin >> n >> s ;
    rep(i,n) cin >> A[i] ;
    int m = n/2 ;
    dp[0][0] = 1 ;
    rep(i,m){
        for(auto it : dp[i]){
            ll val = it.first , cnt = it.second ;
            ll a = A[i] , x = A[i] ;
            while(val + a < s){
                dp[i+1][val+a] += cnt ;
                a *= x ;
            }
        }
    }
    for(auto it : dp[m]){
        ll val = it.first , cnt = it.second ;
        while(cnt > 0){
            X.push_back(val) ;
            cnt-- ;
        }
    }

    ep[0][0] = 1 ;
    rep(i,n-m){
        for(auto it : ep[i]){
            ll val = it.first , cnt = it.second ;
            ll a = A[i+m] , x = A[i+m] ;
            while(val + a < s){
                ep[i+1][val+a] += cnt ;
                a *= x ;
            }
        }
    }
    ll res = 0 ;
    for(auto it : ep[n-m]){
        ll val = it.first , cnt = it.second ;
        auto kt = upper_bound(X.begin(),X.end(),s-val) ;
        res += (ll)(kt - X.begin()) * cnt ;
    }
    cout << res << endl ;
}
0