結果

問題 No.2406 Difference of Coordinate Squared
ユーザー momoyuu
提出日時 2023-10-23 13:12:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 713 ms / 2,000 ms
コード長 1,659 bytes
コンパイル時間 1,232 ms
コンパイル使用メモリ 119,344 KB
実行使用メモリ 85,300 KB
最終ジャッジ日時 2024-09-22 10:06:03
合計ジャッジ時間 19,398 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 55
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<stack>
#include<vector>
#include<set>

using namespace std;
using ll = long long;

#include<atcoder/modint>
using mint = atcoder::modint998244353;


const int mx = 2e6 + 10;
mint fac[mx],ifac[mx];
mint comb(int n,int r){
    return fac[n] * ifac[r] * ifac[n-r];
}

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    ll n,m;
    cin>>n>>m;
    fac[0] = 1;
    for(int i = 1;i<mx;i++) fac[i] = fac[i-1] * i;
    for(int i = 0;i<mx;i++) ifac[i] = fac[i].inv();

    mint ans = 0;
    vector<pair<ll,ll>> use;
    if(m==0){
        for(int i = 1;i<=n;i++) {
            use.push_back(make_pair(0,i));
            use.push_back(make_pair(0,-i));
            use.push_back(make_pair(i,0));
            use.push_back(make_pair(-i,0));
        }
        use.push_back(make_pair(0,0));
    }
    for(ll i = 1;i*i<=abs(m);i++){
        if(m%i!=0) continue;
        ll a = i;
        ll b = m / i;
        use.push_back(make_pair(a,b));
        use.push_back(make_pair(-a,-b));
        swap(a,b);
        use.push_back(make_pair(a,b));
        use.push_back(make_pair(-a,-b));
    }
    sort(use.begin(),use.end());
    use.erase(unique(use.begin(),use.end()),use.end());
    for(int i = 0;i<use.size();i++){
        auto now = use[i];
        ll a = now.first;
        ll b = now.second;
        a = abs(a);
        b = abs(b);
        ll all = n;
        if(n%2!=a%2) continue;
        if(n%2!=b%2) continue;
        ll x = (a+n) / 2;
        ll y = (b+n) / 2;
        if(x>n||y>n) continue;
        ans += comb(n,x) * comb(n,y);
    }
    ans /= mint(4).pow(n);
    cout<<ans.val()<<endl;
    
}
0