Submission #25619059

---

Source Code Expand

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

mint fac(int n){
    static vector<mint> cache{1};
    while(cache.size()<=n)
        cache.push_back(cache.back()*cache.size());
    return cache[n];
}

mint ifac(int n){
    static vector<mint> cache{1};
    while(cache.size()<=n)
        cache.push_back(cache.back()/cache.size());
    return cache[n];
}

mint perm(int n,int m){
    if(n<m)return 0;
    return fac(n)*ifac(n-m);
}

int main(){
    int n,m;
    cin>>n>>m;
    int q=n/m,r=n%m;
    vector<mint> g(n+1);
    for(int k=1;k<=n;k++){
        g[k]=perm(k,q).pow(m-r)*perm(k,q+1).pow(r);
        g[k]*=ifac(k);
    }
    vector<mint> h(n+1);
    for(int i=0;i<=n;i++){
        h[i]=ifac(i);
        if(i%2)h[i]=-h[i];
    }
    vector<mint> f=convolution(g, h);
    for(int i=1;i<=n;i++)
        cout<<f[i].val()<<'\n';
}

Submission Info

Compile Error

./Main.cpp: In function ‘mint fac(int)’:
./Main.cpp:10:23: warning: comparison of integer expressions of different signedness: ‘std::vector<atcoder::static_modint<998244353> >::size_type’ {aka ‘long unsigned int’} and ‘int’ [-Wsign-compare]
   10 |     while(cache.size()<=n)
      |           ~~~~~~~~~~~~^~~
./Main.cpp: In function ‘mint ifac(int)’:
./Main.cpp:17:23: warning: comparison of integer expressions of different signedness: ‘std::vector<atcoder::static_modint<998244353> >::size_type’ {aka ‘long unsigned int’} and ‘int’ [-Wsign-compare]
   17 |     while(cache.size()<=n)
      |           ~~~~~~~~~~~~^~~

Judge Result