Submission #3267197


Source Code Expand

#include<iomanip>
#include<limits>
#include<thread>
#include<utility>
#include<iostream>
#include<string>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<numeric>
#include<cassert>
#include<random>
#include<chrono>
#include<unordered_map>
#include<fstream>
#include<list>
#include<functional>
using namespace std;
typedef unsigned long long int ull;
typedef long long int ll;
typedef pair<ll,ll> pll;
typedef pair<int,int> pi;
typedef pair<double,double> pd;
typedef pair<double,ll> pdl;
#define F first
#define S second
const ll E=1e18+7;
const ll MOD=998244353; /*1000000007;*/





class comb{
private:
    ll mod;
    ll mx;
    vector<ll> F;
    
public:
    comb(ll mod=1000000007,ll mx=100000):mod(mod),mx(mx){
        mk_F();
    }
    
    ll mod_pow(ll a,ll x){
        a%=mod;
        ll ans=1;
        for(int i=0;i<63;i++){
            if(x>>i&1){ans*=a; ans%=mod;}
            a*=a;
            a%=mod;
        }
        return ans;
    }
    
    pll Ex_gcd(ll a,ll b){
        if(b==0){return {1,0};}
        pll ret=Ex_gcd(b,a%b);
        ret.F-=a/b*ret.S;
        return {ret.S,ret.F};
    }
    
    ll prime_R(ll a){
        return mod_pow(a,mod-2);
    }
    
    ll R(ll a){
        ll ret=Ex_gcd(a,mod).F;
        ret%=mod;
        if(ret<0){ret+=mod;}
        return ret;
    }
    
    void mk_F(){
        F={1};
        ll a=1;
        for(ll i=1;i<=mx;i++){
            a*=i;
            a%=mod;
            F.push_back(a);
        }
    }
    
    ll c(ll n,ll k){
        if(n<k){return 0;}
        if(n==k || k==0){return 1;}
        return F[n]*R(F[n-k])%mod*R(F[k])%mod;
    }
    
    //mod must be prime
    ll Lucas_C(ll n,ll m){
        ll ret=1;
        while(n>0 || m>0){
            ret*=c(n%mod,m%mod);
            ret%=mod;
            n/=mod; m/=mod;
        }
        return ret;
    }
    
    ll Stirling(ll n,ll k){
        ll ret=0;
        for(ll i=1;i<=k;i++){
            if((k-i)%2){ret-=c(k,i)*mod_pow(i,n)%mod;}
            else{ret+=c(k,i)*mod_pow(i,n)%mod;}
            ret%=mod;
        }
        ret*=R(F[k]);
        ret%=mod;
        if(ret<0){ret+=mod;}
        return ret;
    }
    
    ll Bell(ll n,ll k){
        ll ret=0;
        for(ll i=1;i<=k;i++){ret+=Stirling(n,i); ret%=mod;}
        return ret;
    }
};

comb C(998244353);

vector<vector<ll>> a(2500,vector<ll>(2500,0));

ll count(ll n,ll k,ll I){
    ll ret=a[n][k];
    ll c=0;
    for(int i=1;i+i<=I;i++){
        if(I-i<=k){c++;}
    }
    for(int i=1;i<=c && i*2<n;i++){
        if(i%2==1){
            ret-=C.c(c,i)*a[n-2*i][k]%MOD;
            ret%=MOD;
        }
        else{
            ret+=C.c(c,i)*a[n-2*i][k]%MOD;
            ret%=MOD;
        }
    }
    ret%=MOD;
    if(ret<0){ret+=MOD;}
    return ret;
}


int main(){
    ll k,n;
    cin>>k>>n;
    for(int i=1;i<=n;i++){
        ll sum=0;
        for(int t=1;t<=k;t++){
            if(i==1){a[i][t]=t; continue;}
            sum+=a[i-1][t];
            sum%=MOD;
            a[i][t]=sum;
        }
    }
    /*
    for(int i=1;i<=k;i++){
        for(int t=1;t<=n;t++){
            cout<<a[t][i]<<" ";
        }
        cout<<endl;
    }
     */
    for(int i=2;i<=2*k;i++){
        cout<<count(n,k,i)<<endl;
    }
    
    
    return 0;
}

Submission Info

Submission Time
Task E - Stop. Otherwise...
User tubuann
Language C++14 (GCC 5.4.1)
Score 0
Code Size 3485 Byte
Status WA
Exec Time 1675 ms
Memory 50164 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 700
Status
AC × 3
AC × 14
WA × 9
Set Name Test Cases
Sample s1.txt, s2.txt, s3.txt
All 01.txt, 02.txt, 03.txt, 04.txt, 05.txt, 06.txt, 07.txt, 08.txt, 09.txt, 10.txt, 11.txt, 12.txt, 13.txt, 14.txt, 15.txt, 16.txt, 17.txt, 18.txt, 19.txt, 20.txt, s1.txt, s2.txt, s3.txt
Case Name Status Exec Time Memory
01.txt WA 1675 ms 50164 KB
02.txt AC 1355 ms 50036 KB
03.txt WA 1382 ms 50036 KB
04.txt AC 25 ms 50036 KB
05.txt AC 36 ms 50036 KB
06.txt WA 1320 ms 50036 KB
07.txt AC 1669 ms 50036 KB
08.txt AC 244 ms 50036 KB
09.txt AC 25 ms 50036 KB
10.txt AC 36 ms 50036 KB
11.txt WA 33 ms 50036 KB
12.txt AC 25 ms 50036 KB
13.txt WA 146 ms 50036 KB
14.txt WA 1333 ms 50036 KB
15.txt WA 1468 ms 50164 KB
16.txt AC 1315 ms 50036 KB
17.txt AC 25 ms 50036 KB
18.txt WA 25 ms 50036 KB
19.txt AC 25 ms 50036 KB
20.txt WA 25 ms 50036 KB
s1.txt AC 25 ms 50036 KB
s2.txt AC 25 ms 50036 KB
s3.txt AC 25 ms 50036 KB