結果

問題 No.824 Many Shifts Hard
ユーザー beetbeet
提出日時 2019-04-27 13:57:51
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 244 ms / 2,000 ms
コード長 2,098 bytes
コンパイル時間 2,116 ms
コンパイル使用メモリ 202,132 KB
実行使用メモリ 225,268 KB
最終ジャッジ日時 2024-05-05 19:37:42
合計ジャッジ時間 7,565 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 153 ms
225,132 KB
testcase_01 AC 153 ms
225,148 KB
testcase_02 AC 165 ms
225,152 KB
testcase_03 AC 179 ms
225,072 KB
testcase_04 AC 169 ms
224,956 KB
testcase_05 AC 224 ms
225,152 KB
testcase_06 AC 155 ms
225,104 KB
testcase_07 AC 166 ms
225,152 KB
testcase_08 AC 213 ms
225,136 KB
testcase_09 AC 192 ms
225,152 KB
testcase_10 AC 154 ms
225,128 KB
testcase_11 AC 160 ms
225,024 KB
testcase_12 AC 176 ms
225,096 KB
testcase_13 AC 208 ms
225,180 KB
testcase_14 AC 244 ms
225,088 KB
testcase_15 AC 230 ms
224,996 KB
testcase_16 AC 160 ms
225,092 KB
testcase_17 AC 242 ms
225,268 KB
testcase_18 AC 155 ms
225,152 KB
testcase_19 AC 238 ms
225,136 KB
testcase_20 AC 237 ms
225,076 KB
testcase_21 AC 156 ms
225,140 KB
testcase_22 AC 239 ms
225,140 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using Int = long long;
template<typename T1,typename T2> inline void chmin(T1 &a,T2 b){if(a>b) a=b;}
template<typename T1,typename T2> inline void chmax(T1 &a,T2 b){if(a<b) a=b;}


template<typename T,T MOD = 1000000007>
struct Mint{
  T v;
  Mint():v(0){}
  Mint(signed v):v(v){}
  Mint(long long t){v=t%MOD;if(v<0) v+=MOD;}

  Mint pow(long long k){
    Mint res(1),tmp(v);
    while(k){
      if(k&1) res*=tmp;
      tmp*=tmp;
      k>>=1;
    }
    return res;
  }
  
  Mint inv(){return pow(MOD-2);}
  
  Mint& operator+=(Mint a){v+=a.v;if(v>=MOD)v-=MOD;return *this;}
  Mint& operator-=(Mint a){v+=MOD-a.v;if(v>=MOD)v-=MOD;return *this;}
  Mint& operator*=(Mint a){v=1LL*v*a.v%MOD;return *this;}
  Mint& operator/=(Mint a){return (*this)*=a.inv();}
  
  Mint operator+(Mint a) const{return Mint(v)+=a;};
  Mint operator-(Mint a) const{return Mint(v)-=a;};
  Mint operator*(Mint a) const{return Mint(v)*=a;};
  Mint operator/(Mint a) const{return Mint(v)/=a;};

  Mint operator-(){return v?MOD-v:v;}

  bool operator==(const Mint a)const{return v==a.v;}
  bool operator!=(const Mint a)const{return v!=a.v;}
  bool operator <(const Mint a)const{return v <a.v;}
};

//INSERT ABOVE HERE
const int MAX = 305;
using M = Mint<int>;
M dp[2][MAX][MAX][MAX];
signed main(){
  int n,k;
  cin>>n>>k;
  
  // x != 1
  dp[0][0][0][1]=1;
  // x == 1
  dp[1][0][0][1]=1;
  
  for(int l=0;l<k;l++){
    for(int i=0;i<=k;i++){
      for(int j=i+1;j<=k+1;j++){
        if(i+1<j){
          dp[0][l+1][i+1][j]+=dp[0][l][i][j];
          dp[1][l+1][i+1][j]+=dp[1][l][i][j];
        }
        dp[0][l+1][i][j+1]+=dp[0][l][i][j];
        dp[1][l+1][i][j+1]+=dp[0][l][i][j];

        dp[0][l+1][i][j]+=dp[0][l][i][j]*M(n-2);   
        dp[1][l+1][i][j]+=dp[1][l][i][j]*M(n-1);  
      }
    }
  }
  
  M ans(0);
  {
    int l=k;
    for(int i=0;i<=k;i++){
      for(int j=i+1;j<=k+1;j++){
        if(j>n) break;
        ans+=dp[0][l][i][j]*M(1LL*((j-i+1)+(n-i))*(n-j)/2);
        ans+=dp[1][l][i][j]*M(j-i);        
      }
    }
  }  
  cout<<ans.v<<endl;
  return 0;
}
0