結果
問題 | No.1102 Remnants |
ユーザー | penta |
提出日時 | 2020-07-11 17:16:03 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 254 ms / 2,000 ms |
コード長 | 3,134 bytes |
コンパイル時間 | 2,001 ms |
コンパイル使用メモリ | 212,272 KB |
実行使用メモリ | 32,768 KB |
最終ジャッジ日時 | 2024-10-13 06:17:54 |
合計ジャッジ時間 | 5,661 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 2 ms
5,248 KB |
testcase_05 | AC | 13 ms
5,248 KB |
testcase_06 | AC | 5 ms
5,248 KB |
testcase_07 | AC | 254 ms
32,768 KB |
testcase_08 | AC | 2 ms
5,248 KB |
testcase_09 | AC | 3 ms
5,248 KB |
testcase_10 | AC | 2 ms
5,248 KB |
testcase_11 | AC | 6 ms
5,248 KB |
testcase_12 | AC | 9 ms
5,248 KB |
testcase_13 | AC | 2 ms
5,248 KB |
testcase_14 | AC | 11 ms
5,248 KB |
testcase_15 | AC | 9 ms
5,248 KB |
testcase_16 | AC | 23 ms
5,248 KB |
testcase_17 | AC | 213 ms
28,800 KB |
testcase_18 | AC | 220 ms
28,928 KB |
testcase_19 | AC | 69 ms
11,904 KB |
testcase_20 | AC | 19 ms
6,008 KB |
testcase_21 | AC | 126 ms
18,432 KB |
testcase_22 | AC | 181 ms
25,216 KB |
testcase_23 | AC | 139 ms
19,968 KB |
testcase_24 | AC | 95 ms
15,104 KB |
testcase_25 | AC | 219 ms
29,312 KB |
testcase_26 | AC | 11 ms
5,248 KB |
testcase_27 | AC | 51 ms
9,984 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; using ll = long long; #define rep(i,n) for (int i = 0; i < (n); ++i) #define rep2(i,m,n) for (int i = m; i < (n); ++i) #define all(x) (x).begin(),(x).end() inline int popcount(const int x) { return __builtin_popcount(x);} template<class T> void chmin(T &a, const T &b) noexcept { if (b < a) a = b;} template<class T> void chmax(T &a, const T &b) noexcept { if (a < b) a = b;} template<class T> void drop(const T &x) { std::cout<<x<<endl; exit(0);} void debug_out() { std::cout << "\n";} template<class T, class... Args> void debug_out(const T &x, const Args &... args) { std::cout<<x<< " "; debug_out(args...);} #ifdef _DEBUG #define debug(...) debug_out(__VA_ARGS__) #else #define debug(...) #endif struct InitIO{ InitIO() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << std::fixed << std::setprecision(15); } }init_io; const int mod = 1000000007; // const int mod = 998244353; struct mint { ll x; mint(ll x=0):x((x%mod+mod)%mod){} mint operator-() const { return mint(-x);} mint& operator+=(const mint a) { if ((x += a.x) >= mod) x -= mod; return *this; } mint& operator-=(const mint a) { if ((x += mod-a.x) >= mod) x -= mod; return *this; } mint& operator*=(const mint a) { (x *= a.x) %= mod; return *this;} mint operator+(const mint a) const { return mint(*this) += a;} mint operator-(const mint a) const { return mint(*this) -= a;} mint operator*(const mint a) const { return mint(*this) *= a;} mint pow(ll t) const { if (!t) return 1; mint a = pow(t>>1); a *= a; if (t&1) a *= *this; return a; } // for prime mod mint inv() const { return pow(mod-2);} mint& operator/=(const mint a) { return *this *= a.inv();} mint operator/(const mint a) const { return mint(*this) /= a;} bool operator==(const mint rhs) const { return x == rhs.x; } bool operator!=(const mint rhs) const { return x != rhs.x; } bool operator<(const mint &a) const{ return x<a.x;}; }; istream& operator>>(istream& is, mint& a) { return is >> a.x;} ostream& operator<<(ostream& os, const mint& a) { return os << a.x;} template<typename ModInt> struct Combination { map<pair<ll,ll>, mint> ncr_mp; Combination(){} ModInt C(ll n, ll r) { if (r < 0 || n < r) return ModInt(0); auto key = pair<ll,ll>(n,r); if (ncr_mp.count(key)) return ncr_mp[key]; if (r == 0 or r == n) return ModInt(1); if (r == 1 or r == n-1) return ModInt(n); //return mp[key] = nCr(n,r-1) * (n-r+1) / r; return ncr_mp[key] = C(n-1,r) * n / (n-r); } ModInt H(ll n, ll r) { if (r < 0 || n < 0) return 0; return r == 0 ? ModInt(1) : C(n+r-1, r); } ModInt operator()(ll n, ll k) { return C(n, k);} }; int main() { ll n, k; cin >> n >> k; vector<ll> a(n); rep(i,n) cin >> a[i]; mint ans = 0; Combination<mint> comb; rep(i, n) { //i番目が残るようなlの選び方はi+1個からk個,rの選び方はn-i個からk個取るパターン数と同じ. ans += comb.H(i+1, k) * comb.H(n-(i+1)+1, k) * mint(a[i]); } cout << ans << endl; return 0; }