結果

問題 No.762 PDCAパス
コンテスト
ユーザー yuppe19 😺
提出日時 2018-12-10 13:34:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 45 ms / 2,000 ms
コード長 961 bytes
コンパイル時間 1,163 ms
コンパイル使用メモリ 84,824 KB
実行使用メモリ 10,240 KB
最終ジャッジ日時 2024-09-16 15:49:22
合計ジャッジ時間 3,294 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge6
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cmath>
#include <iostream>
#include <vector>
using namespace std;
using i64 = int64_t;

vector<i64> dp;
vector<vector<int>> G;
vector<int> arr;
constexpr int mod = static_cast<int>(powl(10, 9)) + 7;

i64 rec(int v, int cur) {
  i64 &res = dp[v];
  if(cur == 3) { return 1; }
  if(res != -1) { return res; }
  res = 0;
  for(int u : G[v]) {
    if(arr[u] == cur+1) {
      res += rec(u, cur+1);
      res %= mod;
    }
  }
  return res;
}

int main(void) {
  int n, m; scanf("%d%d", &n, &m);
  string s; cin >> s;
  arr.resize(n);
  for(int i=0; i<n; ++i) {
    arr[i] = static_cast<int>(string("PDCA").find(s[i]));
  }
  G.assign(n, vector<int>());
  for(int i=0; i<m; ++i) {
    int u, v; scanf("%d%d", &u, &v);
    --u, --v;
    G[u].push_back(v);
    G[v].push_back(u);
  }
  dp.assign(n, -1);
  i64 res = 0;
  for(int i=0; i<n; ++i) {
    if(arr[i] == 0) {
      res += rec(i, 0);
      res %= mod;
    }
  }
  printf("%ld\n", res);
  return 0;
}
0