提出 #35641840

---

ソースコード 拡げる

#include <bits/stdc++.h>
#include <atcoder/modint>
#include <atcoder/convolution>
using mint = atcoder::modint998244353;
using namespace std;

typedef long long ll;
#define rep(i,a,n) for (ll i=a;i<(ll)n;i++)
ll read(){ll r;scanf("%lld",&r);return r;}

int n;

vector<vector<vector<mint>>> a; // [idx][奇偶][value] = 方案数

vector<vector<mint>> calc(int x,int y){
  auto r00 = atcoder::convolution(a[x][0],a[y][0]);
  auto r11 = atcoder::convolution(a[x][1],a[y][1]);

  auto r01 = atcoder::convolution(a[x][0],a[y][1]);
  auto r10 = atcoder::convolution(a[x][1],a[y][0]);

  r00.resize(max(r00.size(),r11.size()));
  rep(i,0,r11.size()) r00[i]+=r11[i];

  r01.resize(max(r01.size(),r10.size()));
  rep(i,0,r10.size()) r01[i]+=r10[i];
  return {r00,r01};
}

int main(){
  int n=read();
  int m=read();
  int n2=1;
  while(n2<n)n2*=2;
  a=vector(n2,vector(2,vector<mint>(1,0)));
  rep(i,0,n){
    int x=read();
    a[i][1].resize(x+1);
    a[i][1][x]=1;
  }
  rep(i,0,n2)a[i][0][0]=1;
  for(;n2>1;n2/=2){
    vector<vector<vector<mint>>> nexta;
    rep(i,0,n2/2) nexta.emplace_back(calc(i*2,i*2+1));
    a=nexta;
  }
  printf("%d\n",m<(int)a[0][1].size()?a[0][1][m].val():0);
  return 0;
}

提出情報

コンパイルエラー

./Main.cpp: In function ‘ll read()’:
./Main.cpp:9:21: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
    9 | ll read(){ll r;scanf("%lld",&r);return r;}
      |                ~~~~~^~~~~~~~~~~

ジャッジ結果