結果

問題 No.762 PDCAパス
ユーザー tetsuzuki1115
提出日時 2019-08-26 02:27:20
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 74 ms / 2,000 ms
コード長 1,591 bytes
コンパイル時間 1,239 ms
コンパイル使用メモリ 101,740 KB
実行使用メモリ 8,960 KB
最終ジャッジ日時 2024-11-07 09:30:38
合計ジャッジ時間 3,842 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <cstring>
#include <math.h>
#include <cmath>
#include <limits.h>
#include <map>
#include <unordered_map>
#include <set>
#include <queue>
#include <algorithm>
#include <functional>
#include <stdio.h>
using namespace std;

long long MOD = 1000000007;

vector< vector<int> > P;
string S;
vector< long long > memo;


bool is_PDCA( char a, char b ) {
    bool ret = false;
    string pdca = "PDCA";
    for ( int i = 0; i < 3; i++ ) {
        if ( pdca[i] == a && pdca[i+1] == b ) { ret = true; }
    }
    return ret;
}

long long dfs( int v ) {

    if ( memo[v] != -1 ) { return memo[v]; }
    if ( S[v] == 'A' ) { return 1; }

    long long ret = 0;
    for ( int i = 0; i < P[v].size(); i++ ) {
        ret += dfs( P[v][i] );
        ret %= MOD;
    }
    memo[v] = ret;
    return ret;
}

int main() {

    int N,M;
    cin >> N >> M >> S;
    
    P.resize(N);
    memo.resize(N);
    for ( int i = 0; i < N; i++ ) {
        memo[i] = -1;
    }

    vector<int> start;
    for ( int i = 0; i < S.length(); i++ ) {
        if ( S[i] == 'P' ) { start.push_back(i); }
    }

    for ( int i = 0; i < M; i++ ) {
        int a,b;
        cin >> a >> b;
        a--;
        b--;
        if ( is_PDCA( S[a], S[b] ) ) {
            P[a].push_back( b );
        }
        if ( is_PDCA( S[b], S[a] ) ) {
            P[b].push_back( a );
        }        
    }

    long long ans = 0;
    for ( int i = 0; i < start.size(); i++ ) {
        ans += dfs( start[i] );
        ans %= MOD;
    }

    cout << ans << endl;


    return 0;
}
0