結果
問題 | No.1035 Color Box |
ユーザー |
![]() |
提出日時 | 2020-04-27 18:51:13 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 58 ms / 2,000 ms |
コード長 | 1,828 bytes |
コンパイル時間 | 1,151 ms |
コンパイル使用メモリ | 108,616 KB |
実行使用メモリ | 6,824 KB |
最終ジャッジ日時 | 2024-11-22 03:15:14 |
合計ジャッジ時間 | 3,853 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 36 |
ソースコード
#pragma GCC optimize("O3")#include <iostream>#include <iomanip>#include <cstdio>#include <string>#include <cstring>#include <deque>#include <list>#include <queue>#include <stack>#include <vector>#include <utility>#include <algorithm>#include <map>#include <set>#include <complex>#include <cmath>#include <limits>#include <cfloat>#include <climits>#include <ctime>#include <cassert>#include <numeric>#include <fstream>#include <functional>#include <bitset>using namespace std;using ll = long long;using P = pair<int, int>;using T = tuple<int, int, int>;template <class T> inline T chmax(T &a, const T b) {return a = (a < b) ? b : a;}template <class T> inline T chmin(T &a, const T b) {return a = (a > b) ? b : a;}constexpr int MOD = 1e9 + 7;constexpr int inf = 1e9;constexpr long long INF = 1e18;int dx[] = {1, 0, -1, 0};int dy[] = {0, 1, 0, -1};ll modpow(ll a, ll b){if(b == 0) return 1;else if(b % 2 == 0){ll d = modpow(a, b / 2) % MOD;return (d * d) % MOD;}else{return (a * modpow(a, b - 1)) % MOD;}}const int MAX_N = 100010;ll fact[MAX_N], finv[MAX_N];ll comb(int n, int r){if(n < r || r < 0) return 0;return fact[n] * finv[n-r] % MOD * finv[r] % MOD;}void comb_init(int n){fact[0] = finv[0] = 1;for(int i=1; i<=n; i++){fact[i] = (fact[i-1] * i) % MOD;finv[i] = modpow(fact[i], MOD - 2);}}int main(){cin.tie(0);ios::sync_with_stdio(false);int n, k; cin>>n>>k;comb_init(100005);ll ans = 0;for(int i=0; i<=k; i++){int rest = k - i;ll cur = comb(k, i) * modpow(rest, n) % MOD;if(i % 2 == 0) ans = (ans + cur) % MOD;else ans = (ans - cur + MOD) % MOD;}cout << ans << endl;return 0;}