Submission #25239792

---

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 binom(int n,int m){
    return fac(n)*ifac(m)*ifac(n-m);
}

int main(){
    int n;
    cin>>n;
    vector<int> col(n);
    for(int &x:col)cin>>x;
    sort(col.begin(),col.end());
    vector<int> cnt{1};
    for(int i=1;i<n;i++)
        if(col[i]!=col[i-1])
            cnt.push_back(1);
        else
            cnt.back()++;
    vector<vector<mint>> f,fd;
    priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> pq;
    for(int c:cnt){
        pq.push({c,f.size()});
        vector<mint> unit(c+1);
        for(int i=0;i<=c;i++)
            unit[i]=binom(c,i);
        f.push_back(unit);
        fd.push_back(unit);
        fd.back()[0]=0;
    }
    while(pq.size()>1){
        int i,sizei,j,sizej;
        tie(sizei,i)=pq.top();
        pq.pop();
        tie(sizej,j)=pq.top();
        pq.pop();
        vector<mint> f_new=convolution(f[i],f[j]),
                     fd_new=convolution(f[i],fd[j]),
                     fd_new_2=convolution(fd[i],f[j]);
        for(int i=0;i<f_new.size();i++)
            fd_new[i]+=fd_new_2[i];
        f[i]=f_new;
        fd[i]=fd_new;
        f[j]={};
        fd[j]={};
        pq.push({sizei+sizej,i});
    }
    int ians=pq.top().second;
    for(int i=1;i<f[ians].size();i++)
        cout<<(fd[ians][i]/f[ians][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)
      |           ~~~~~~~~~~~~^~~
./Main.cpp: In function ‘int main()’:
./Main.cpp:58:22: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<atcoder::static_modint<998244353> >::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   58 |         for(int i=0;i<f_new.size();i++)
      |                     ~^~~~~~~~~~~~~
./Main.cpp:67:18: warning: comparison of integer expressions of different signedness: ‘int’ and ‘std::vector<atcoder::static_modint<998244353> >::size_type’ {aka ‘long unsigned int’} [-Wsign-compare]
   67 |     for(int i=1;i<f[ians].size();i++)
      |                 ~^~~~~~~~~~~~~~~

Judge Result