結果

問題 No.3403 Count 1210 Sequence
コンテスト
ユーザー Mike scarf
提出日時 2025-12-12 15:55:19
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 1,612 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 2,996 ms
コンパイル使用メモリ 276,604 KB
実行使用メモリ 10,144 KB
最終ジャッジ日時 2025-12-12 15:55:53
合計ジャッジ時間 33,214 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample WA * 1
other WA * 31
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:26:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   26 |     scanf("%lld",&t);
      |     ~~~~~^~~~~~~~~~~
main.cpp:28:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   28 |         scanf("%lld%lld",&n,&a);
      |         ~~~~~^~~~~~~~~~~~~~~~~~

ソースコード

diff #
raw source code

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll mod=998244353;
const int maxn=400005;
ll fac[maxn+5],inv[maxn+5];
ll qmksm(ll a,ll b){
    ll res=1;
    while(b){
        if(b&1) res=res*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return res;
}
ll C(ll n,ll k){
    if(k<0 or k>n) return 0;
    return fac[n]*inv[k]%mod*inv[n-k]%mod;
}
int main(){
    ll t,n,a,ans,i,j,k,zc,zc1,zc2;
    fac[0]=1;
    for(i=1;i<=maxn;i++) fac[i]=fac[i-1]*i%mod;
    inv[maxn]=qmksm(fac[maxn],mod-2);
    for(i=maxn-1;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
    scanf("%lld",&t);
    while(t--){
        scanf("%lld%lld",&n,&a);
        if(n==1){
            if(a==0) printf("1\n");
            else printf("0\n");
            continue;
        }
        ans=0;
        for(k=1;k*k<=a;k++){
            if(a%k==0){
                zc=k;
                if((a/zc)%2==(n-1)%2){
                    zc1=(n-1-(a/zc))/2;
                    if(zc1>=0 and zc1<=n-1) ans=(ans+C(n-1,zc1))%mod;
                }
                if(k*k!=a){
                    zc=a/k;
                    if((a/zc)%2==(n-1)%2){
                        zc1=(n-1-(a/zc))/2;
                        if(zc1>=0 and zc1<=n-1) ans=(ans+C(n-1,zc1))%mod;
                    }
                }
            }
        }
        if(a==0){
            ans=0;
            zc1=(n-1)/2;
            if((n-1)%2==0) ans=C(n-1,zc1);
        }
        ans=(ans*2)%mod;
        if(a==0) ans=ans*qmksm(2,mod-2)%mod;
        if(a%(n-1)==0) ans=(ans+2)%mod;
        if(a==0) ans=(ans-1+mod)%mod;
        printf("%lld\n",ans);
    }
    return 0;
}
0