結果
問題 | No.117 組み合わせの数 |
ユーザー | FF256grhy |
提出日時 | 2017-10-24 15:04:12 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 3,174 bytes |
コンパイル時間 | 1,455 ms |
コンパイル使用メモリ | 169,488 KB |
実行使用メモリ | 34,808 KB |
最終ジャッジ日時 | 2024-11-21 17:36:33 |
合計ジャッジ時間 | 3,297 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
コンパイルメッセージ
In file included from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/istream:39, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/sstream:38, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/complex:45, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/ccomplex:39, from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/x86_64-pc-linux-gnu/bits/stdc++.h:54, from main.cpp:1: In member function 'std::basic_ostream<_CharT, _Traits>::__ostream_type& std::basic_ostream<_CharT, _Traits>::operator<<(long long int) [with _CharT = char; _Traits = std::char_traits<char>]', inlined from 'int main()' at main.cpp:105:11: /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/ostream:202:25: warning: 'ans' may be used uninitialized [-Wmaybe-uninitialized] 202 | { return _M_insert(__n); } | ~~~~~~~~~^~~~~ main.cpp: In function 'int main()': main.cpp:100:20: note: 'ans' was declared here 100 | LL ans; | ^~~
ソースコード
#include <bits/stdc++.h> using namespace std; typedef long long signed int LL; typedef long long unsigned int LU; #define incID(i, l, r) for(int i = (l) ; i < (r); i++) #define incII(i, l, r) for(int i = (l) ; i <= (r); i++) #define decID(i, l, r) for(int i = (r) - 1; i >= (l); i--) #define decII(i, l, r) for(int i = (r) ; i >= (l); i--) #define inc(i, n) incID(i, 0, n) #define inc1(i, n) incII(i, 1, n) #define dec(i, n) decID(i, 0, n) #define dec1(i, n) decII(i, 1, n) #define inII(v, l, r) ((l) <= (v) && (v) <= (r)) #define inID(v, l, r) ((l) <= (v) && (v) < (r)) #define PB push_back #define EB emplace_back #define MP make_pair #define FI first #define SE second #define PQ priority_queue #define ALL(v) v.begin(), v.end() #define RALL(v) v.rbegin(), v.rend() #define FOR(it, v) for(auto it = v.begin(); it != v.end(); ++it) #define RFOR(it, v) for(auto it = v.rbegin(); it != v.rend(); ++it) template<typename T> bool setmin(T & a, T b) { if(b < a) { a = b; return true; } else { return false; } } template<typename T> bool setmax(T & a, T b) { if(b > a) { a = b; return true; } else { return false; } } template<typename T> bool setmineq(T & a, T b) { if(b <= a) { a = b; return true; } else { return false; } } template<typename T> bool setmaxeq(T & a, T b) { if(b >= a) { a = b; return true; } else { return false; } } template<typename T> T gcd(T a, T b) { return (b == 0 ? a : gcd(b, a % b)); } template<typename T> T lcm(T a, T b) { return a / gcd(a, b) * b; } // ---- ---- LL MOD; LL mod(LL x, LL m = MOD) { return (x % m + m) % m; } pair<LL, LL> ex_gcd(LL a, LL b) { if(b == 0) { return MP(1, 0); } auto p = ex_gcd(b, a % b); return MP(p.SE, p.FI - (a / b) * p.SE); } LL inv(LL x, LL m = MOD) { assert(gcd(x, m) == 1); auto p = ex_gcd(x, m); return mod(p.FI, m); } LL promod(LL x, LL y, LL m = MOD) { return mod((x % m) * (y % m), m); } LL divmod(LL x, LL y, LL m = MOD) { return promod(x, inv(y, m), m); } // ---- struct CombMod { LL lim = 0, mod; LL * fact; // fact[i]: i の階乗 LL * finv; // finv[i]: i の階乗の逆元 CombMod() { } CombMod(LL lim, LL mod = MOD) { init(lim, mod); } void init(LL arg_lim, LL arg_mod = MOD) { lim = arg_lim; mod = arg_mod; fact = new LL[lim + 1]; finv = new LL[lim + 1]; fact[0] = 1; inc1(i, lim) { fact[i] = promod(fact[i - 1], i, mod); } finv[lim] = inv(fact[lim], mod); dec(i, lim) { finv[i] = promod(finv[i + 1], i + 1, mod); } } LL P(LL a, LL b) { assert(inII(a, 0, lim) && inII(b, 0, lim)); return (a < b ? 0 : promod(fact[a], finv[a - b], mod)); } LL C(LL a, LL b) { assert(inII(a, 0, lim) && inII(b, 0, lim)); return (a < b ? 0 : promod(P(a, b), finv[b], mod)); } LL H(LL a, LL b) { assert(inII(a, 0, lim) && inII(b, 0, lim) && inII(a + b - 1, 0, lim)); return (a == 0 ? 0 : C(a + b - 1, b)); } }; // ---- int main() { MOD = 1e9 + 7; CombMod cm(2000000); int t, n, k; char c; scanf("%d\n", &t); inc(i, t) { scanf("%c(%d,%d)\n", &c, &n, &k); LL ans; if(c == 'P') { ans = cm.P(n, k); } if(c == 'C') { ans = cm.C(n, k); } if(c == 'H') { ans = cm.H(n, k); } cout << ans << "\n"; } return 0; }