結果

問題 No.1693 Invasion
ユーザー uytvcc
提出日時 2021-10-02 02:01:34
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 51 ms / 2,000 ms
コード長 1,891 bytes
コンパイル時間 1,580 ms
コンパイル使用メモリ 172,144 KB
実行使用メモリ 16,216 KB
最終ジャッジ日時 2024-07-19 19:50:24
合計ジャッジ時間 3,455 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using ull = unsigned long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)
#define MOD 1000000007
#define MOD2 998244353
#define INF 1000000007
#define LINF 1000000000000000007LL
#define PI 3.14159265359
#define P pair<ll,ll>
template <typename T>
inline bool chmax(T &a, const T &b){
  if(a<b) {a=b; return true;}
  else return false;
}
template <typename T>
inline bool chmin(T &a, const T &b){
  if(a>b) {a=b; return true;}
  else return false;
}
struct Edge{
  int to; ll cost;
  Edge(int to, ll cost) : to(to), cost(cost) {}
};
typedef vector<Edge> Edges;
typedef vector<Edges> Graph;
void add_edge(Graph &g,int from,int to,ll cost,bool rev,ll rev_cost){
  g[from].push_back(Edge(to,cost));
  if(rev) g[to].push_back(Edge(from,rev_cost));
}

ll dp[100005];
int a[110];

const int MAX = 510000; // 5*(10^5)

ll fac[MAX], finv[MAX], inv[MAX];

void init(int m) {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++){
        fac[i] = fac[i - 1] * i % m;
        inv[i] = m - inv[m%i] * (m / i) % m;
        finv[i] = finv[i - 1] * inv[i] % m;
    }
}

ll nCk(int n, int k,int m){
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % m) % m;
}

ll nPk(int n, int k, int m){
    if(n < k) return 0;
    if(n < 0 || k < 0) return 0;
    return fac[n]*finv[n-k]%m;
}

void solve(){
  int n,m;
  cin>>n>>m;
  rep(i,n) cin>>a[i];

  rep(i,m+1) dp[i]=INF;
  dp[0]=0;
  for(int k=0; k<=m; k++)rep(i,n){
    if(k-a[i]>=0) chmin(dp[k],dp[k-a[i]]+1); 
  }
  
  init(MOD2);
  ll ans=0;
  for(int k=0; k<=m; k++){
    if(dp[k]!=INF) ans += nCk(m-dp[k],k-dp[k],MOD2);
    ans %=MOD2;
  }
  cout<<ans<<'\n';
}

int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  cout<<setprecision(10)<<fixed;
  solve();
  return 0;
}
0