結果
問題 | No.829 成長関数インフレ中 |
ユーザー |
![]() |
提出日時 | 2019-05-03 22:43:14 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 219 ms / 2,000 ms |
コード長 | 2,626 bytes |
コンパイル時間 | 1,180 ms |
コンパイル使用メモリ | 124,936 KB |
最終ジャッジ日時 | 2025-01-07 03:27:31 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 22 |
ソースコード
#include<iostream> #include<iomanip> #include<vector> #include<algorithm> #include<set> #include<map> #include<unordered_map> #include<stack> #include<queue> #include<math.h> #include<functional> #include<bitset> #include<cassert> using namespace std; using lint = long long; using ld = long double; using pint = pair<int, int>; using plint = pair<lint, lint>; #define MOD 1000000007LL #define INF 1000000000LL #define EPS 1e-10 #define FOR(i,n,m) for(lint i=n;i<(int)m;i++) #define REP(i,n) FOR(i,0,n) #define DUMP(a) REP(d,a.size()){cout<<a[d];if(d!=a.size()-1)cout<<" ";else cout<<endl;} #define ALL(v) v.begin(),v.end() #define UNIQUE(v) sort(ALL(v));v.erase(unique(ALL(v)),v.end()); #define pb push_back #define MAX_N 1000000 long long extgcd(long long a, long long b, long long& x, long long& y) { long long d = a; if (b != 0) { d = extgcd(b, a % b, y, x); y -= (a / b) * x; } else { x = 1; y = 0; } return d; } long long mod_inverse(long long a, long long m) { long long x, y; if(extgcd(a, m, x, y) == 1) return (m + x % m) % m; else return -1; } vector<long long> fact(MAX_N+1, INF); long long mod_fact(long long n, long long& e) { if(fact[0] == INF) { fact[0]=1; if(MAX_N != 0) fact[1]=1; for(lint i = 2; i <= MAX_N; ++i) { fact[i] = (fact[i-1] * i) % MOD; } } e = 0; if(n == 0) return 1; long long res = mod_fact(n / MOD, e); e += n / MOD; if((n / MOD) % 2 != 0) return (res * (MOD - fact[n % MOD])) % MOD; return (res * fact[n % MOD]) % MOD; } // return nCk long long mod_comb(long long n, long long k) { if(n < 0 || k < 0 || n < k) return 0; long long e1, e2, e3; long long a1 = mod_fact(n, e1), a2 = mod_fact(k, e2), a3 = mod_fact(n - k, e3); if(e1 > e2 + e3) return 0; return (a1 * mod_inverse((a2 * a3) % MOD, MOD)) % MOD; } int main() { ios::sync_with_stdio(false); cin.tie(0); // factorial vector<lint> fact(202020); fact[0] = 1; FOR(i, 1, 202020) { fact[i] = i * fact[i - 1]; fact[i] %= MOD; } lint n, b; cin >> n >> b; map<int, int> m; REP(i, n) { int s; cin >> s; m[s]++; } vector<int> v; for(auto pr: m) v.pb(pr.second); reverse(ALL(v)); lint val1 = 0, val2 = 1; int sum = 0; REP(i, v.size()) { lint c1 = 0, c2 = 0; c1 = mod_comb(sum + v[i] - 1, v[i]) * fact[v[i]]; c1 %= MOD; c2 = mod_comb(sum + v[i] - 1, sum) * fact[v[i]]; c2 %= MOD; lint nval1 = 0, nval2 = 0; nval1 = c1 * val1 + c2 * (((val1 + val2) * b) % MOD); nval1 %= MOD; nval2 = c1 * val2 + c2 * ((val2 * b) % MOD); nval2 %= MOD; val1 = nval1; val2 = nval2; sum += v[i]; } cout << val1 << endl; return 0; } /* --------------------------------------- */