#469843 (C++17) No.1035 Color Box

提出ソース

結果

問題	No.1035 Color Box
ユーザー	r1933
提出日時	2020-04-24 22:55:49
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	WA
実行時間	-
コード長	7,107 bytes
コンパイル時間	2,141 ms
コンパイル使用メモリ	199,548 KB
実行使用メモリ	15,360 KB
最終ジャッジ日時	2024-04-23 04:03:41
合計ジャッジ時間	3,541 ms
ジャッジサーバーID （参考情報）	judge4 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	20 ms 15,168 KB
testcase_01	WA	-
testcase_02	WA	-
testcase_03	WA	-
testcase_04	AC	14 ms 15,356 KB
testcase_05	WA	-
testcase_06	WA	-
testcase_07	WA	-
testcase_08	AC	20 ms 15,172 KB
testcase_09	WA	-
testcase_10	WA	-
testcase_11	AC	14 ms 15,284 KB
testcase_12	WA	-
testcase_13	AC	19 ms 15,136 KB
testcase_14	AC	18 ms 15,140 KB
testcase_15	AC	20 ms 15,068 KB
testcase_16	AC	14 ms 15,004 KB
testcase_17	AC	14 ms 15,184 KB
testcase_18	AC	14 ms 15,296 KB
testcase_19	AC	20 ms 15,176 KB
testcase_20	AC	15 ms 15,148 KB
testcase_21	AC	15 ms 15,180 KB
testcase_22	AC	14 ms 15,132 KB
testcase_23	AC	13 ms 15,224 KB
testcase_24	AC	14 ms 15,140 KB
testcase_25	AC	15 ms 15,060 KB
testcase_26	WA	-
testcase_27	AC	14 ms 15,248 KB
testcase_28	AC	14 ms 15,300 KB
testcase_29	AC	14 ms 15,164 KB
testcase_30	WA	-
testcase_31	AC	16 ms 15,068 KB
testcase_32	AC	15 ms 15,216 KB
testcase_33	AC	15 ms 15,064 KB
testcase_34	AC	14 ms 15,224 KB
testcase_35	WA	-
testcase_36	WA	-
testcase_37	AC	16 ms 15,360 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include "bits/stdc++.h"

// Begin Header {{{
using namespace std;

#ifndef DEBUG
#define dump(...)
#endif

#define all(x) x.begin(), x.end()
#define rep(i, b, e) for (intmax_t i = (b), i##_limit = (e); i < i##_limit; ++i)
#define reps(i, b, e) for (intmax_t i = (b), i##_limit = (e); i <= i##_limit; ++i)
#define repr(i, b, e) for (intmax_t i = (b), i##_limit = (e); i >= i##_limit; --i)
#define var(Type, ...) Type __VA_ARGS__; input(__VA_ARGS__)

constexpr size_t    operator""_zu(unsigned long long value) { return value; };
constexpr intmax_t  operator""_jd(unsigned long long value) { return value; };
constexpr uintmax_t operator""_ju(unsigned long long value) { return value; };

constexpr int INF = 0x3f3f3f3f;
constexpr intmax_t LINF = 0x3f3f3f3f3f3f3f3f_jd;

template <class T, class Compare = less<>>
using MaxHeap = priority_queue<T, vector<T>, Compare>;
template <class T, class Compare = greater<>>
using MinHeap = priority_queue<T, vector<T>, Compare>;

inline void input() {}
template <class Head, class... Tail>
inline void input(Head&& head, Tail&&... tail) {
    cin >> head;
    input(forward<Tail>(tail)...);
}

template <class T>
inline istream& operator>>(istream &is, vector<T> &vec) {
    for (auto &e: vec) {
        is >> e;
    }
    return is;
}

inline void output() { cout << "\n"; }
template <class Head, class... Tail>
inline void output(Head&& head, Tail&&... tail) {
    cout << head;
    if (sizeof...(tail)) {
        cout << " ";
    }
    output(forward<Tail>(tail)...);
}

template <class T>
inline ostream& operator<<(ostream &os, const vector<T> &vec) {
    static constexpr const char *delim[] = {" ", ""};
    for (const auto &e: vec) {
        os << e << delim[&e == &vec.back()];
    }
    return os;
}

template <class T>
inline vector<T> makeVector(const T &initValue, size_t sz) {
    return vector<T>(sz, initValue);
}

template <class T, class... Args>
inline auto makeVector(const T &initValue, size_t sz, Args... args) {
    return vector<decltype(makeVector<T>(initValue, args...))>(sz, makeVector<T>(initValue, args...));
}

template <class Func>
class FixPoint : Func {
public:
    explicit constexpr FixPoint(Func&& f) noexcept : Func(forward<Func>(f)) {}

    template <class... Args>
    constexpr decltype(auto) operator()(Args&&... args) const {
        return Func::operator()(*this, std::forward<Args>(args)...);
    }
};

template <class Func>
static inline constexpr decltype(auto) makeFixPoint(Func&& f) noexcept {
    return FixPoint<Func>{forward<Func>(f)};
}

template <class Container>
struct reverse_t {
    Container &c;
    reverse_t(Container &c) : c(c) {}
    auto begin() { return c.rbegin(); }
    auto end() { return c.rend(); }
};

template <class Container>
auto reversed(Container &c) {
    return reverse_t<Container>(c);
}

template <class T>
inline bool chmax(T &a, const T &b) noexcept {
    return b > a && (a = b, true);
}

template <class T>
inline bool chmin(T &a, const T &b) noexcept {
    return b < a && (a = b, true);
}

template <class T>
inline T diff(const T &a, const T &b) noexcept {
    return a < b ? b - a : a - b;
}
// End Header }}}

const intmax_t MOD = intmax_t(1e9) + 7;

// ModInt {{{
template <intmax_t MOD>
class ModInt {
    intmax_t value;

public:
    inline ModInt(const ModInt& other) :
        value(other.value)
    {}

    inline ModInt(intmax_t value = 0) {
        if (value >= MOD) {
            this->value = value % MOD;
        } else if (value < 0) {
            this->value = (value + MOD) % MOD;
        } else {
            this->value = value;
        }
    }

    template <class T>
        explicit inline operator T() const {
            return static_cast<T>(value);
        }

    inline ModInt inverse() const {
        return ModInt::pow(value, MOD - 2);
    }

    inline ModInt& operator+=(const ModInt other) {
        value = (value + other.value) % MOD;
        return *this;
    }

    inline ModInt& operator-=(const ModInt other) {
        value = (MOD + value - other.value) % MOD;
        return *this;
    }

    inline ModInt& operator*=(const ModInt other) {
        value = (value * other.value) % MOD;
        return *this;
    }

    inline ModInt& operator/=(const ModInt other) {
        value = (value * other.inverse().value) % MOD;
        return *this;
    }

    inline ModInt operator+(const ModInt other) {
        return ModInt(*this) += other;
    }

    inline ModInt operator-(const ModInt other) {
        return ModInt(*this) -= other;
    }

    inline ModInt operator*(const ModInt other) {
        return ModInt(*this) *= other;
    }

    inline ModInt operator/(const ModInt other) {
        return ModInt(*this) /= other;
    }

    inline bool operator==(const ModInt other) {
        return value == other.value;
    }

    inline bool operator!=(const ModInt other) {
        return value != other.value;
    }

    friend ostream& operator<<(ostream &os, const ModInt other) {
        os << other.value;
        return os;
    }

    friend istream& operator>>(istream &is, ModInt& other) {
        is >> other.value;
        return is;
    }

    static constexpr inline ModInt pow(intmax_t n, intmax_t p) {
        intmax_t ret = 1;
        for (; p > 0; p >>= 1) {
            if (p & 1) ret = (ret * n) % MOD;
            n = (n * n) % MOD;
        }
        return ret;
    }
};

using Mint = ModInt<MOD>;
// }}}

// Factorial {{{
#ifdef DEBUG
#define constexpr
#endif

template <size_t N, size_t MOD>
struct Factorial {
    vector<uint_fast64_t> fact_;

    constexpr inline Factorial() : fact_(N + 1) {
        fact_[0] = 1;
        for (intmax_t i = 1; i <= N; ++i) {
            fact_[i] = (fact_[i - 1] * i) % MOD;
        }
    }

    constexpr inline Mint operator[](size_t pos) const {
        return fact_[pos];
    }
};

template <size_t N, size_t MOD>
struct InvFact {
    vector<uint_fast64_t> inv_;
    vector<uint_fast64_t> finv_;

    constexpr inline InvFact() : inv_(N + 1), finv_(N + 1) {
        inv_[1] = 1;
        finv_[0] = finv_[1] = 1;
        for (intmax_t i = 2; i <= N; ++i) {
            inv_[i] = (MOD - MOD / i) * inv_[MOD % i] % MOD;
            finv_[i] = finv_[i - 1] * inv_[i] % MOD;
        }
    }

    constexpr inline Mint operator[](size_t pos) const {
        return finv_[pos];
    }
};

#ifdef constexpr
#undef constexpr
#endif

Factorial<510000, MOD> fact;
InvFact<510000, MOD> finv;

inline Mint nCr(int n, int r) {
    if (r < 0 || n < r) {
        return 0;
    }
    return fact[n] * (finv[r] * finv[n - r]);
}

inline Mint nPr(int n, int r) {
    if (r < 0 || n < r) {
        return 0;
    }
    return fact[n] * finv[n - r];
}

inline Mint nHr(int n, int r) {
    return nCr(n + r - 1, r);
}
// }}}

signed main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.setf(ios_base::fixed);
    cout.precision(10);
    var(intmax_t, N, M);
    Mint res = Mint::pow(M, N);
    Mint u = 0;
    rep(i, 1, M) {
        const Mint val = nCr(M, i) * Mint::pow(i, N);
        if (i & 1) u += val;
        else u -= val;
    }
    output(res - u);
    return 0;
}

yukicoder

結果

テストケース

ソースコード