結果

問題 No.3635 Probability trip
コンテスト
ユーザー yuunegi
提出日時 2026-08-21 22:49:59
言語 C++23
(gcc 15.2.0 + boost 1.90.0)
コンパイル:
g++-15 -O2 -lm -std=c++23 -Wuninitialized -DONLINE_JUDGE -o a.out _filename_
実行:
./a.out
結果
WA  
実行時間 -
コード長 1,495 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,070 ms
コンパイル使用メモリ 389,688 KB
実行使用メモリ 9,412 KB
最終ジャッジ日時 2026-08-21 22:50:19
合計ジャッジ時間 7,054 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 2
other AC * 8 WA * 35
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
#if __has_include("all.hpp")
#include "all.hpp"
#else
#include <atcoder/all>
#endif
using namespace atcoder;
using mint=modint998244353;
int main(void){
    int n,m;
    cin>>n>>m;
    vector<vector<int>>v(n);
    for(int i=0;i<m;i++){
        int a,b;
        cin>>a>>b;
        a--,b--;
        v[b].push_back(a);
        v[a].push_back(b);
    }
    auto ml = [&](vector<vector<mint>>A,vector<vector<mint>>B){
        vector<vector<mint>>tmp(n,vector<mint>(n,0));
        for(int i=0;i<n;i++)for(int j=0;j<n;j++)for(int k=0;k<n;k++)tmp[i][j]+=A[k][i]*B[j][k];
        return tmp;
    };
    vector<vector<vector<mint>>>doub(62,vector<vector<mint>>(n,vector<mint>(n,0)));
    for(int i=0;i<n;i++){
        if(v[i].size()==0)continue;
        for(int j=0;j<v[i].size();j++){
            doub[0][i][v[i][j]]=1/(mint)(v[i].size());
        }
    }
    for(int i=1;i<62;i++){
        doub[i]=ml(doub[i-1],doub[i-1]);
    }
    long long a,b,s,t;
    cin>>s>>t>>a>>b;
    a--,b--;
    auto calc = [&](long long startt,long long goalt,long long startp,long long goalp){
        vector<vector<mint>>tmp(n,vector<mint>(n,0));
        for(int i=0;i<n;i++)tmp[i][i]=1;
        long long cnt=goalt-startt;
        for(long long i=61;i>=0;i--){
            if((1<<i)&cnt){
                tmp=ml(tmp,doub[i]);
            }
        }
        return tmp[startp][goalp];
    };
    cout<<(calc(1,t,0,b)*calc(t,s,b,a)/calc(1,s,0,a)).val()<<endl;
    return 0;
}
0