結果
問題 | No.823 Many Shifts Easy |
ユーザー | Im_Gimlet |
提出日時 | 2020-07-23 22:28:12 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 21 ms / 2,000 ms |
コード長 | 2,652 bytes |
コンパイル時間 | 1,744 ms |
コンパイル使用メモリ | 171,432 KB |
実行使用メモリ | 19,032 KB |
最終ジャッジ日時 | 2024-06-23 21:15:41 |
合計ジャッジ時間 | 2,454 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 19 ms
18,832 KB |
testcase_01 | AC | 18 ms
19,032 KB |
testcase_02 | AC | 18 ms
18,856 KB |
testcase_03 | AC | 21 ms
18,816 KB |
testcase_04 | AC | 18 ms
19,028 KB |
testcase_05 | AC | 20 ms
18,944 KB |
testcase_06 | AC | 20 ms
18,988 KB |
testcase_07 | AC | 20 ms
18,948 KB |
testcase_08 | AC | 19 ms
18,852 KB |
testcase_09 | AC | 20 ms
18,848 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return true; } return false; } template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return true; } return false; } using ll = long long; using P = pair<ll, ll>; const long double PI = acos(-1.0L); ll GCD(ll a, ll b) { return b?GCD(b, a%b):a; } ll LCM(ll a, ll b) { return a/GCD(a, b)*b; } // auto mod int const int mod = 1000000007; // const int mod = 998244353; struct mint { ll x; // typedef long long ll; 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;} }; istream& operator>>(istream& is, mint& a) { return is >> a.x;} ostream& operator<<(ostream& os, const mint& a) { return os << a.x;} // combination mod prime // mintと併用 struct combination { vector<mint> fact, ifact; combination(int n):fact(n+1),ifact(n+1) { assert(n < mod); fact[0] = 1; for (int i = 1; i <= n; ++i) fact[i] = fact[i-1]*i; ifact[n] = fact[n].inv(); for (int i = n; i >= 1; --i) ifact[i-1] = ifact[i]*i; } mint operator()(int n, int k) { if (k < 0 || k > n) return 0; return fact[n]*ifact[k]*ifact[n-k]; } mint p(int n, int k) { if(n < k) return 0; return fact[n]*ifact[n-k]; } } comb(1000005); int main() { int n, k; cin >> n >> k; mint ans = 0; for(int i = 1; i <= n; ++i) { mint sum = 0; if(i == n) { // 1番右の要素は選ばれないときだけ sum += comb.p(n-1, k); }else { // それ以外は選ばれないor選ばれるけどi+1も選ばれる sum += comb.p(n-1, k); mint ch = comb(n-2, k-2); // iとi+1以外のk-2個を選ぶ組み合わせ sum += (ch*comb.p(k, k-2)); } ans += (sum*(mint)i); } cout << ans << endl; }