結果

問題 No.2457 Stampaholic (Easy)
コンテスト
ユーザー Nachia
提出日時 2023-08-25 04:32:06
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,680 bytes
コンパイル時間 740 ms
コンパイル使用メモリ 78,648 KB
最終ジャッジ日時 2025-02-16 13:25:28
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <atcoder/modint>
using namespace std;
using i32 = int;
using u32 = unsigned int;
using i64 = long long;
using u64 = unsigned long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)
const i64 INF = 1001001001001001001;

using Modint = atcoder::static_modint<998244353>;

int main(){
    i64 H, W, N, K; cin >> H >> W >> N >> K;
    i64 f = (H-K+1) * (W-K+1);
    Modint p = (Modint(f)).inv().pow(N);
    Modint ans = 0;
    i64 cnt = 0;
    for(i64 x=1; x<K && x*2-1<=W; x++) for(i64 y=1; y<K && y*2-1<=H; y++){
        Modint tmp = Modint(f-min(x,W-K+1)*min(y,H-K+1)).pow(N);
        int dcnt = 1;
        if(x*2 <= W) tmp += tmp, dcnt += dcnt;
        if(y*2 <= H) tmp += tmp, dcnt += dcnt;
        ans += tmp, cnt += dcnt;
    }
    if(W >= 2*K-2){
        for(int y=1; y<K && y*2-1<=H; y++){
            Modint tmp = Modint(f-K*y).pow(N) * (W-(2*K-2));
            ans += tmp, cnt += (W-(2*K-2));
            if(y*2 <= H) ans += tmp, cnt += (W-(2*K-2));
        }
    }
    if(H >= 2*K-2){
        for(int x=1; x<K && x*2-1<=W; x++){
            Modint tmp = Modint(f-K*x).pow(N) * (H-(2*K-2));
            ans += tmp, cnt += (H-(2*K-2));
            if(x*2 <= W) ans += tmp, cnt += (H-(2*K-2));
        }
    }
    if(W >= 2*K-2 && H >= 2*K-2){
        ans += Modint(f-K*K).pow(N) * (H-(2*K-2)) * (W-(2*K-2)), cnt += (H-(2*K-2)) * (W-(2*K-2));
    }
    ans *= p;
    ans = Modint(H*W) - ans;
    cout << ans.val() << endl;
    return 0;
}


struct ios_do_not_sync{
    ios_do_not_sync(){
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0