結果

問題 No.1049 Zero (Exhaust)
ユーザー walkrewalkre
提出日時 2020-05-08 22:06:59
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 27 ms / 2,000 ms
コード長 3,518 bytes
コンパイル時間 1,386 ms
コンパイル使用メモリ 170,944 KB
実行使用メモリ 18,900 KB
最終ジャッジ日時 2024-07-04 00:45:17
合計ジャッジ時間 2,511 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 1 ms
6,940 KB
testcase_03 AC 20 ms
13,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 17 ms
11,888 KB
testcase_06 AC 27 ms
18,848 KB
testcase_07 AC 5 ms
6,944 KB
testcase_08 AC 11 ms
7,680 KB
testcase_09 AC 21 ms
14,840 KB
testcase_10 AC 23 ms
15,848 KB
testcase_11 AC 23 ms
16,068 KB
testcase_12 AC 25 ms
18,036 KB
testcase_13 AC 8 ms
6,940 KB
testcase_14 AC 14 ms
9,884 KB
testcase_15 AC 21 ms
13,996 KB
testcase_16 AC 18 ms
11,880 KB
testcase_17 AC 17 ms
12,268 KB
testcase_18 AC 24 ms
16,108 KB
testcase_19 AC 20 ms
14,216 KB
testcase_20 AC 11 ms
7,808 KB
testcase_21 AC 21 ms
14,540 KB
testcase_22 AC 14 ms
9,580 KB
testcase_23 AC 27 ms
18,736 KB
testcase_24 AC 27 ms
18,900 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using i64 = int64_t;

#define rep(i, x, y) for (i64 i = i64(x), i##_max_for_repmacro = i64(y); i < i##_max_for_repmacro; ++i)
#define debug(x) #x << "=" << (x)

#ifdef DEBUG
#define _GLIBCXX_DEBUG
#define print(x) std::cerr << debug(x) << " (L:" << __LINE__ << ")" << std::endl
#else
#define print(x)
#endif

template <i64 p>
class fp {
    public:
    i64 x;
    fp() : x(0) {}
    fp(i64 x_) : x((x_ % p + p) % p) {}
    fp operator+() const { return fp(x); }
    fp operator-() const { return fp(-x); }
    fp& operator+=(const fp& y) {
        x += y.x;
        if (x >= p) x -= p;
        return *this;
    }
    fp& operator-=(const fp& y) { return *this += -y; }
    fp& operator*=(const fp& y) {
        x = x * y.x % p;
        return *this;
    }
    fp& operator/=(const fp& y) { return *this *= fp(inverse(y.x)); }
    fp operator+(const fp& y) const { return fp(x) += y; }
    fp operator-(const fp& y) const { return fp(x) -= y; }
    fp operator*(const fp& y) const { return fp(x) *= y; }
    fp operator/(const fp& y) const { return fp(x) /= y; }
    bool operator==(const fp& y) const { return x == y.x; }
    bool operator!=(const fp& y) const { return !(*this == y); }
    i64 extgcd(i64 a, i64 b, i64& x, i64& y) {
        i64 d = a;
        if (b != 0) {
            d = extgcd(b, a % b, y, x);
            y -= (a / b) * x;
        } else {
            x = 1;
            y = 0;
        }
        return d;
    }
    i64 inverse(i64 a) {
        i64 x, y;
        extgcd(a, p, x, y);
        return (x % p + p) % p;
    }
};

template <i64 p>
i64 abs(const fp<p>& x) { return x.x; }

template <i64 p>
istream& operator>>(istream& is, fp<p>& x) {
    is >> x.x;
    return is;
}

template <i64 p>
ostream& operator<<(ostream& os, const fp<p>& x) {
    os << x.x;
    return os;
}

template <typename T, typename U>
ostream& operator<<(ostream& os, const pair<T, U>& p) {
    os << "(" << p.first << ", " << p.second << ")";
    return os;
}

template <typename T>
ostream& operator<<(ostream& os, const vector<T>& vec) {
    os << "[";
    for (const auto& v : vec) {
        os << v << ",";
    }
    os << "]";
    return os;
}

template <typename T>
bool chmin(T& a, const T& b) {
    if (a > b) {
        a = b;
        return true;
    }
    return false;
}

template <typename T>
bool chmax(T& a, const T& b) {
    if (a < b) {
        a = b;
        return true;
    }
    return false;
}

template <typename A, typename T, size_t size>
void fill(A (&ary)[size], const T& val) {
    fill((T*)ary, (T*)(ary + size), val);
}

constexpr int inf = 1.01e9;
constexpr i64 inf64 = 4.01e18;
constexpr long double eps = 1e-9;

// double(64bit浮動小数)のn分探索のループ回数の上限(2分探索なら50でも十分かもしれない). long double(80ビットの x87 浮動小数点型?)だと, 2分探索であってもこれだと足りないケースがある気がするので, もうちょっと余裕を持たせた方が良さそう.
constexpr i64 max_loop = 100;

void solve() {
    constexpr i64 mod = 1'000'000'007;

    i64 P,K;
    cin >> P >> K;

    vector<fp<mod>> all(K),cnt(K);
    all[0]=2*P;
    cnt[0]=1+P;
    rep(i,1,K){
        all[i]=all[i-1]*(2*P);
        cnt[i]+=cnt[i-1]*(1+P);
        cnt[i]+=(all[i-1]-cnt[i-1])*(1+1);
    }

    cout << cnt[K-1] << endl;
}

int main() {
    std::cin.tie(0);
    std::ios::sync_with_stdio(false);
    cout.setf(ios::fixed);
    cout.precision(16);
    solve();
    return 0;
}
0