結果
問題 | No.2151 3 on Torus-Lohkous |
ユーザー |
![]() |
提出日時 | 2022-12-15 09:34:52 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 147 ms / 2,000 ms |
コード長 | 896 bytes |
コンパイル時間 | 3,248 ms |
コンパイル使用メモリ | 227,056 KB |
最終ジャッジ日時 | 2025-02-09 13:44:36 |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 8 |
ソースコード
#include<bits/stdc++.h> #include<atcoder/modint> #include<atcoder/convolution> using namespace std; using lint = long long; #define rep(i,n) for(lint i=0;i<(n);++i) using mint=atcoder::static_modint<998244353>; int main(){ vector<mint>table(1000000,1),inv(1000000); rep(i,1000000)table[i+1]=table[i]*(i+1); inv[999999]=table[999999].inv(); for(lint i=999998;i>=0;--i)inv[i]=inv[i+1]*(i+1); auto comb=[&](auto a,auto b){ return table[a]*inv[b]*inv[a-b]; }; lint t; cin>>t; while(t--){ lint h,w; cin>>h>>w; lint n=gcd(h,w); mint ans=mint(h)*w; if(n>3){ ans+=2*n; } if(h*w/n%3==0){ if(n>4)ans+=n*12; } if(h%4==0&&w%4==0){ ans+=16; } if(n%2==0){ rep(p,2){ mint x=0; for(lint i=(n+3*p+5)%6+1;i<=n;i+=6){ if(0<=n/2-5*i){ x+=comb((n/2-5*i)/3+i-1,i-1)/i; } } ans+=x*x*n*n; } } cout<<ans.val()<<endl; } }