結果
問題 |
No.824 Many Shifts Hard
|
ユーザー |
![]() |
提出日時 | 2020-04-19 03:51:45 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
TLE
(最新)
AC
(最初)
|
実行時間 | - |
コード長 | 1,971 bytes |
コンパイル時間 | 4,309 ms |
コンパイル使用メモリ | 157,436 KB |
最終ジャッジ日時 | 2025-01-09 21:22:57 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 18 TLE * 5 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #define popcount __builtin_popcount using namespace std; //typedef long long int ll; typedef pair<int, int> P; using ll=unsigned long long; const ll MOD=1e9+7; ll comb[303][303]; ll dp[303][303][303]; ll s[303][303]; int c[303][303]; int main() { int n, k; cin>>n>>k; for(int i=0; i<=k; i++){ comb[i][0]=1, comb[i][i]=1; for(int j=1; j<i; j++) comb[i][j]=(comb[i-1][j]+comb[i-1][j-1])%MOD; } s[1][0]=1; for(int i=2; i<=k+1; i++){ dp[i][1][1]=i-1; dp[i][1][2]=1; for(int j=2; j<=k; j++){ for(int m=1; m<j; m++){ for(int l=2; l<=min(i, m+2); l++){ dp[i][j][l]+=dp[i-1][m][l-1]*comb[j-1][m]; if(!(c[j][l]&15)) dp[i][j][l]%=MOD; c[j][l]++; } } } for(int j=2; j<=k; j++) dp[i][j][2]++; for(int j=2; j<=k; j++){ for(int l=1; l<=min(i, j+1); l++){ if(l==1){ (dp[i][j][l]+=dp[i][j-1][l]*(i-1)+dp[i][j-1][l+1])%=MOD; }else{ (dp[i][j][l]+=dp[i][j-1][l]*(i-2)+dp[i][j-1][l+1])%=MOD; } } } s[i][0]=1; for(int j=1; j<=k; j++) s[i][j]=dp[i][j][1]; } ll ans=0; for(int i=1; i<=n; i++){ int m=min(n-i+1, k+1); ll p=1; for(int j=0; j<=k; j++){ (ans+=comb[k][j]*p%MOD*i%MOD*s[m][k-j])%=MOD; (p*=(n-m))%=MOD; } } cout<<ans<<endl; return 0; }