結果

問題 No.1474 かさまＪ
ユーザー 👑 NachiaNachia
提出日時 2021-04-10 00:32:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 275 ms / 2,500 ms
コード長 1,291 bytes
コンパイル時間 2,257 ms
コンパイル使用メモリ 206,420 KB
最終ジャッジ日時 2025-01-20 15:23:19
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:28:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   28 |   scanf("%d%d%d%d",&N,&P,&Q,&L);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:29:17: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   29 |   rep(i,N) scanf("%d",&S[i]);
      |            ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using ll = long long;
using ull = unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

const ll MOD = 1000000007;
using mll = atcoder::static_modint<1000000007>;

int N,P,Q,L;
int S[40];
vector<vector<mll>> dp;

vector<mll> F,I,iF;
mll C(int n,int r){ return F[n]*iF[r]*iF[n-r]; }

void initC(int Z){
  F.resize(Z+1);
  F[0]=1; for(int i=1; i<=Z; i++) F[i]=F[i-1]*i;
  I.resize(Z+1);
  I[1]=1; for(int i=2; i<=Z; i++) I[i]=-(MOD/i*I[MOD%i]);
  iF.resize(Z+1);
  iF[0]=1; for(int i=1; i<=Z; i++) iF[i]=iF[i-1]*I[i];
}

int main(){
  scanf("%d%d%d%d",&N,&P,&Q,&L);
  rep(i,N) scanf("%d",&S[i]);
  dp.assign(N+1,vector<mll>(Q+1,0));
  dp[0][0]=1;
  rep(i,N){
    vector<vector<mll>> buf = dp;
    rep(j,N+1) rep(q,Q) dp[j][q+1] += dp[j][q];
    rep(j,N) for(int q=1; q<=Q; q++){
      int l = max(0,q-S[i]);
      int r = q-1;
      if(l==0) buf[j+1][q] += dp[j][r];
      else buf[j+1][q] += dp[j][r] - dp[j][l-1];
    }
    dp = move(buf);
  }

  initC(40000);

  mll ans=0;

  for(int i=0; i<=N; i++){
    if(i*L>P+Q) break;
    for(int q=0; q<=Q; q++){
      int freeP = P - (i*L-q);
      if(freeP<0) continue;
      ans += dp[i][q] * C(freeP+N-1,N-1);
    }
  }

  printf("%u\n",ans.val());

  return 0;
}
0