結果
問題 | No.762 PDCAパス |
ユーザー |
|
提出日時 | 2018-12-11 09:57:05 |
言語 | D (dmd 2.109.1) |
結果 |
AC
|
実行時間 | 56 ms / 2,000 ms |
コード長 | 1,710 bytes |
コンパイル時間 | 948 ms |
コンパイル使用メモリ | 108,680 KB |
実行使用メモリ | 9,640 KB |
最終ジャッジ日時 | 2024-06-13 02:12:58 |
合計ジャッジ時間 | 3,528 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 38 |
ソースコード
import std.stdio, std.string, std.conv; import std.range, std.algorithm, std.array, std.typecons, std.container; import std.math, std.numeric, core.bitop; enum mod = 10^^9 + 7; void main() { int[char] cnv = ['P':0, 'D':1, 'C':2, 'A':3]; int n, m; scan(n, m); auto s = readln.chomp.map!(ch => cnv[ch.to!char]).array; auto adj = new int[][](n, 0); foreach (i ; 0 .. m) { int ui, vi; scan(ui, vi); ui--, vi--; if (s[ui] > s[vi]) { swap(ui, vi); } if (s[vi] - s[ui] == 1) { adj[vi] ~= ui; } } auto dp = new int[](n); dp[] = -1; int dfs(int v) { if (s[v] == 0) { return dp[v] = 1; } if (dp[v] != -1) { return dp[v]; } dp[v] = 0; foreach (u ; adj[v]) { dp[v] += dfs(u); dp[v] %= mod; } return dp[v]; } int ans; foreach (i ; 0 .. n) { if (dp[i] == -1) { dfs(i); } if (s[i] == 3) { ans += dp[i]; ans %= mod; } } writeln(ans); } void scan(T...)(ref T args) { import std.stdio : readln; import std.algorithm : splitter; import std.conv : to; import std.range.primitives; auto line = readln().splitter(); foreach (ref arg; args) { arg = line.front.to!(typeof(arg)); line.popFront(); } assert(line.empty); } void fillAll(R, T)(ref R arr, T value) { static if (is(typeof(arr[] = value))) { arr[] = value; } else { foreach (ref e; arr) { fillAll(e, value); } } }