結果

問題 No.1840 Random Painting
ユーザー hitonanode
提出日時 2021-12-12 23:52:56
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,721 bytes
コンパイル時間 1,308 ms
コンパイル使用メモリ 112,744 KB
実行使用メモリ 19,040 KB
最終ジャッジ日時 2024-06-27 15:56:45
合計ジャッジ時間 11,827 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 6 WA * 4 TLE * 2 -- * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
using namespace std;

#include <atcoder/modint>
#include <atcoder/math>
using mint1 = atcoder::modint998244353;
using mint2 = atcoder::modint998244353;

template <class Mint> Mint A107984(Mint n, Mint k) { return (k + 1) * (n + 2) * (2 * n - k + 3) * (n - k + 1) / 6; }

// 右に r ステップ進むより左に l ステップ進む方が先の確率 * （そのようなときにそれを達成するまでの時間の条件付き期待値）
template <class Mint> Mint solve(int l, int r) {
    int N = l + r;
    return A107984<Mint>(N - 2, l - 1) * 2 / Mint(N * N);
}

template <class Mint> Mint sol(const string &S) {
    const int N = S.size();
    vector<int> zeropos;
    for (int i = 0; i < N; i++) {
        if (S[i] == '1') continue;
        zeropos.push_back(i);
    }
    mint1 ret = 0;

    for (int t = 0; t < 2; ++t) {
        ret += solve<mint1>(N - zeropos[0], zeropos[0]);
        for (auto &x : zeropos) x = N - x;
        reverse(zeropos.begin(), zeropos.end());
    }

    for (int i = 1; i < int(zeropos.size()); ++i) {
        int a = zeropos[i - 1], b = zeropos[i];
        for (int t = 0; t < 2; ++t) {
            auto e1 = solve<mint1>(a, N - b);
            auto e2 = solve<mint1>(N - (b - a), b - a);
            ret += e1 * (b - a) / N + e2 * (N - b) / (N - b + a);
            swap(a, b);
            a = N - a;
            b = N - b;
        }
    }
    return ret;
}

int main() {
    cin.tie(nullptr), ios::sync_with_stdio(false);
    int N;
    string S;
    cin >> N >> S;
    auto v1 = sol<mint1>(S).val();
    auto v2 = sol<mint2>(S).val();
    cout << atcoder::crt({v1, v2}, {mint1::mod(), mint2::mod()}).first << '\n';
}
0