結果
問題 | No.1552 Simple Dice Game |
ユーザー |
![]() |
提出日時 | 2021-06-19 00:35:56 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 957 ms / 2,500 ms |
コード長 | 1,497 bytes |
コンパイル時間 | 3,827 ms |
コンパイル使用メモリ | 229,888 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-06-22 21:42:40 |
合計ジャッジ時間 | 15,575 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 20 |
ソースコード
#include "bits/stdc++.h"#include "atcoder/all"using namespace std;using namespace atcoder;//using mint = modint1000000007;//const int mod = 1000000007;using mint = modint998244353;const int mod = 998244353;//const int INF = 1e9;//const long long LINF = 1e18;//const bool debug = false;#define rep(i, n) for (int i = 0; i < (n); ++i)#define rep2(i,l,r)for(int i=(l);i<(r);++i)#define rrep(i, n) for (int i = (n-1); i >= 0; --i)#define rrep2(i,l,r)for(int i=(r-1);i>=(l);--i)#define all(x) (x).begin(),(x).end()#define allR(x) (x).rbegin(),(x).rend()#define endl "\n"#define P pair<int,int>template<typename A, typename B> inline bool chmax(A & a, const B & b) { if (a < b) { a = b; return true; } return false; }template<typename A, typename B> inline bool chmin(A & a, const B & b) { if (a > b) { a = b; return true; } return false; }int main() {ios::sync_with_stdio(false);cin.tie(nullptr);long long n; cin >> n;int m; cin >> m;mint ans = 0;rep2(i, 1, m + 1) {mint anssub = 1;anssub += ((mint)i + 1)*i / 2 * pow_mod(i, n - 1, mod);anssub -= ((mint)(i - 1) + 1)*(i - 1) / 2 * pow_mod(i - 1, n - 1, mod);anssub *= i;anssub *= n;ans += anssub;}rep2(i, 1, m + 1) {mint anssub = 1;anssub += ((mint)i + m)*(m - i + 1) / 2 * pow_mod(m - i + 1, n - 1, mod);anssub -= ((mint)(i + 1) + m)*(m - (i + 1) + 1) / 2 * pow_mod(m - (i + 1) + 1, n - 1, mod);anssub *= i;anssub *= n;ans -= anssub;}cout << ans.val() << endl;return 0;}