Submission #3267207


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(ll 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<2500;i++){
        ll sum=0;
        for(int t=1;t<2500;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(ll 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 3487 Byte
Status WA
Exec Time 1683 ms
Memory 50036 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 1683 ms 50036 KB
02.txt AC 1367 ms 50036 KB
03.txt WA 1398 ms 50036 KB
04.txt AC 48 ms 50036 KB
05.txt AC 59 ms 50036 KB
06.txt WA 1337 ms 50036 KB
07.txt AC 1679 ms 50036 KB
08.txt AC 262 ms 50036 KB
09.txt AC 47 ms 50036 KB
10.txt AC 59 ms 50036 KB
11.txt WA 56 ms 50036 KB
12.txt AC 48 ms 50036 KB
13.txt WA 168 ms 50036 KB
14.txt WA 1348 ms 50036 KB
15.txt WA 1482 ms 50036 KB
16.txt AC 1328 ms 50036 KB
17.txt AC 48 ms 50036 KB
18.txt WA 47 ms 50036 KB
19.txt AC 47 ms 50036 KB
20.txt WA 48 ms 50036 KB
s1.txt AC 48 ms 50036 KB
s2.txt AC 48 ms 50036 KB
s3.txt AC 48 ms 50036 KB