結果
問題 | No.2917 二重木 |
ユーザー |
![]() |
提出日時 | 2024-10-06 18:28:18 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 7 ms / 3,000 ms |
コード長 | 904 bytes |
コンパイル時間 | 1,046 ms |
コンパイル使用メモリ | 108,408 KB |
実行使用メモリ | 7,168 KB |
最終ジャッジ日時 | 2024-10-06 18:28:21 |
合計ジャッジ時間 | 2,419 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 35 |
ソースコード
#include<iostream>#include<vector>#include<algorithm>using namespace std;using ll = long long;#include<atcoder/modint>using mint = atcoder::modint;int main(){cin.tie(nullptr);ios::sync_with_stdio(false);ll n,p;cin>>n>>p;mint::set_mod(p);vector<vector<mint>> comb(n+1,vector<mint>(n+1,0));comb[0][0] = 1;for(int i = 1;i<=n;i++){comb[i][i] = comb[i][0] = 1;for(int j = 1;j<n;j++) comb[i][j] = comb[i-1][j-1] + comb[i-1][j];}mint ans = 0;vector<mint> cal(n+1,0);cal[1] = 1;for(int i = 2;i<=n;i++) cal[i] = mint(i).pow(i-2);for(int i = 1;i<=n;i++){mint tmp = comb[n][i];int res = n - i + 1;tmp *= cal[i];if(res==1){ans += tmp;continue;}tmp *= mint(n).pow(res-2) * i;ans += tmp;}cout<<ans.val()<<endl;}