結果
問題 |
No.1929 Exponential Sequence
|
ユーザー |
|
提出日時 | 2022-05-06 23:52:52 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 39 ms / 2,000 ms |
コード長 | 1,711 bytes |
コンパイル時間 | 1,050 ms |
コンパイル使用メモリ | 110,500 KB |
最終ジャッジ日時 | 2025-01-29 04:13:53 |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 24 |
ソースコード
#include <algorithm> #include <cmath> #include <iostream> #include <map> #include <numeric> #include <queue> #include <vector> using namespace std; using ll = long long; #define rep(i, j, n) for (int i = j; i < (n); ++i) #define rrep(i, j, n) for (int i = (n)-1; j <= i; --i) #define all(a) a.begin(), a.end() template <typename T> std::ostream &operator<<(std::ostream &os, std::vector<T> &a) { for (size_t i = 0; i < a.size(); ++i) os << (i > 0 ? " " : "") << a[i]; return os << '\n'; } template <typename T> std::istream &operator>>(std::istream &is, std::vector<T> &a) { for (T &x : a) { is >> x; } return is; } [[maybe_unused]] constexpr long long MOD = 998244353; // constexpr long long MOD = 1000000007; [[maybe_unused]] constexpr int INF = 0x3f3f3f3f; [[maybe_unused]] constexpr long long INFL = 0x3f3f3f3f3f3f3f3fLL; int main() { cin.tie(0)->sync_with_stdio(0); int n, s; cin >> n >> s; vector<int> a(n); cin >> a; int m = (n + 1) / 2; map<ll, ll> mp, pm; // 1 mp[0] = 1; rep(i, 0, m) { map<ll, ll> nmp; for (auto [j, k] : mp) { ll c = a[i]; while (j + c <= s) { nmp[j + c] += k; c *= a[i]; } } mp = nmp; } // 2 pm[0] = 1; rep(i, m, n) { map<ll, ll> nmp; for (auto [j, k] : pm) { ll c = a[i]; while (j + c <= s) { nmp[j + c] += k; c *= a[i]; } } pm = nmp; } vector<ll> A{0}, B{0}; for (auto [k, v] : pm) { A.push_back(k); B.push_back(v); } rep(i, 1, B.size()) B[i] += B[i - 1]; ll ans = 0; for (auto [k, v] : mp) { ll aaa = s - k; ans += v * B[upper_bound(A.begin(), A.end(), aaa) - A.begin() - 1]; } cout << ans; return 0; }