Submission #36571934

---

Source Code Expand

#include <bits/stdc++.h>
#include <atcoder/modint>
const int MOD=998244353;
using mint = atcoder::static_modint<MOD>;
using namespace std;
#define rep(i,a,n) for (int i=a;i<(int)n;i++)
#define per(i,a,n) for (int i=n;i-->(int)a;)
int read(){int r;scanf("%d",&r);return r;}

vector<int> G[3010];
int c[3010];

mint dp[3010][3010][2][2];// dp[i次移动][停止在j]
                          // [0][0] 用于(v,v), [0][1] 用于(v,\neq v), [1][0] 用于(\neq v,v),[1][1]:(\neq v,\neq v)
mint dinv[3010]; // 1/du[i]
int main(){
  int n=read();
  int m=read();
  int k=read();
  rep(i,1,m+1){
    int u=read();
    int v=read();
    G[u].push_back(v);
    G[v].push_back(u);
  }
  rep(i,1,n+1) c[i] = read();
  rep(i,1,n+1) dinv[i] = mint(G[i].size()).inv();

  dp[0][1][0][0] = 1;
  rep(i,0,k) rep(u,1,n+1) rep(a,0,2) rep(b,0,2) for(auto v: G[u]) rep(na,a,2) rep(nb,b,2){ // na >= a, nb >= b
    if(c[v] == 0 || (na==a && nb==b)) dp[i+1][v][na][nb] += dp[i][u][a][b] * dinv[u];
  }
  mint ans = 0;
  rep(i,1,k+1)rep(j,1,n+1) if(c[j]) ans += dp[i][j][1][1];
  printf("%d\n",ans.val());
  return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int read()’:
./Main.cpp:8:23: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
    8 | int read(){int r;scanf("%d",&r);return r;}
      |                  ~~~~~^~~~~~~~~

Judge Result