結果

問題 No.1929 Exponential Sequence
コンテスト
ユーザー cologne
提出日時 2022-05-10 15:26:43
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 125 ms / 2,000 ms
コード長 890 bytes
コンパイル時間 3,008 ms
コンパイル使用メモリ 255,304 KB
実行使用メモリ 13,036 KB
最終ジャッジ日時 2024-09-14 03:55:16
合計ジャッジ時間 4,629 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int N, S;
    cin >> N >> S;
    vector<int> A(N);
    for (int &a : A)
        cin >> a;

    auto m = [&](vector<int> A)
    {
        vector<int> V = {0};
        for (int a : A)
        {
            vector<int> W;
            int x = 1;
            while ((long)x * a <= S)
            {
                x *= a;
                for (int b : V)
                    if (x + b <= S)
                        W.push_back(x + b);
            }
            V = W;
        }
        return V;
    };

    auto X = m(vector<int>(A.begin(), A.begin() + N / 2));
    auto Y = m(vector<int>(A.begin() + N / 2, A.end()));
    sort(X.begin(), X.end());
    sort(Y.begin(), Y.end());
    long ans = 0;
    for(int x: X)
        ans += distance(Y.begin(), upper_bound(Y.begin(), Y.end(), S-x));

    cout << ans << endl;

    return 0;
}
0