結果

問題 No.824 Many Shifts Hard
ユーザー chocoruskchocorusk
提出日時 2020-04-19 03:51:45
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,959 ms / 2,000 ms
コード長 1,971 bytes
コンパイル時間 1,216 ms
コンパイル使用メモリ 131,032 KB
実行使用メモリ 220,920 KB
最終ジャッジ日時 2024-10-04 00:45:13
合計ジャッジ時間 18,343 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 45 ms
28,032 KB
testcase_03 AC 365 ms
93,312 KB
testcase_04 AC 211 ms
63,232 KB
testcase_05 AC 1,327 ms
181,120 KB
testcase_06 AC 9 ms
5,248 KB
testcase_07 AC 114 ms
35,328 KB
testcase_08 AC 1,117 ms
168,192 KB
testcase_09 AC 695 ms
123,136 KB
testcase_10 AC 30 ms
14,336 KB
testcase_11 AC 91 ms
31,488 KB
testcase_12 AC 341 ms
82,560 KB
testcase_13 AC 913 ms
153,728 KB
testcase_14 AC 1,777 ms
220,768 KB
testcase_15 AC 1,510 ms
200,960 KB
testcase_16 AC 53 ms
32,640 KB
testcase_17 AC 1,769 ms
220,352 KB
testcase_18 AC 2 ms
5,248 KB
testcase_19 AC 1,795 ms
220,920 KB
testcase_20 AC 1,942 ms
220,628 KB
testcase_21 AC 3 ms
5,248 KB
testcase_22 AC 1,959 ms
220,428 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#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;
}
0