#326938 (Go) No.762 PDCAパス

提出ソース

結果

問題	No.762 PDCAパス
ユーザー	tsuchinaga
提出日時	2019-03-22 10:57:13
言語	Go (1.22.1)
結果	AC
実行時間	150 ms / 2,000 ms
コード長	1,769 bytes
コンパイル時間	10,832 ms
コンパイル使用メモリ	235,680 KB
実行使用メモリ	14,592 KB
最終ジャッジ日時	2024-09-19 02:17:56
合計ジャッジ時間	13,602 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 5,248 KB
testcase_01	AC	1 ms 5,376 KB
testcase_02	AC	1 ms 5,376 KB
testcase_03	AC	1 ms 5,376 KB
testcase_04	AC	1 ms 5,376 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	1 ms 5,376 KB
testcase_07	AC	1 ms 5,376 KB
testcase_08	AC	1 ms 5,376 KB
testcase_09	AC	2 ms 5,376 KB
testcase_10	AC	2 ms 5,376 KB
testcase_11	AC	1 ms 5,376 KB
testcase_12	AC	1 ms 5,376 KB
testcase_13	AC	1 ms 5,376 KB
testcase_14	AC	1 ms 5,376 KB
testcase_15	AC	1 ms 5,376 KB
testcase_16	AC	1 ms 5,376 KB
testcase_17	AC	1 ms 5,376 KB
testcase_18	AC	1 ms 5,376 KB
testcase_19	AC	1 ms 5,376 KB
testcase_20	AC	1 ms 5,376 KB
testcase_21	AC	1 ms 5,376 KB
testcase_22	AC	14 ms 5,376 KB
testcase_23	AC	14 ms 5,376 KB
testcase_24	AC	149 ms 14,080 KB
testcase_25	AC	150 ms 14,592 KB
testcase_26	AC	16 ms 5,376 KB
testcase_27	AC	15 ms 5,376 KB
testcase_28	AC	130 ms 10,112 KB
testcase_29	AC	135 ms 10,112 KB
testcase_30	AC	112 ms 9,472 KB
testcase_31	AC	111 ms 9,344 KB
testcase_32	AC	107 ms 9,344 KB
testcase_33	AC	78 ms 7,296 KB
testcase_34	AC	79 ms 7,552 KB
testcase_35	AC	78 ms 7,296 KB
testcase_36	AC	38 ms 5,376 KB
testcase_37	AC	38 ms 5,376 KB
testcase_38	AC	38 ms 5,376 KB
testcase_39	AC	38 ms 5,376 KB
testcase_40	AC	40 ms 5,376 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

package main

import (
	"bufio"
	"fmt"
	"math"
	"os"
	"strconv"
)

func main() {
	var n, m int
	s := ""
	_, _ = fmt.Scan(&n, &m, &s)
	r := []rune(s)
	pdca := []rune("PDCA")
	mod := int(math.Pow(10, 9)) + 7

	sr := make(map[int][]int, 0) // PDCAの位置
	for i, c := range r {
		if c == pdca[0] {
			sr[0] = append(sr[0], i)
		} else if c == pdca[1] {
			sr[1] = append(sr[1], i)
		} else if c == pdca[2] {
			sr[2] = append(sr[2], i)
		} else {
			sr[3] = append(sr[3], i)
		}
	}

	sc := bufio.NewScanner(os.Stdin)
	sc.Split(bufio.ScanWords)

	// P -> D, D -> C, C -> Aのパスのみを取得する
	path := make(map[int][]int, 0) // from -> to
	for i := 0; i < m; i++ {
		sc.Scan()
		n1, _ := strconv.Atoi(sc.Text())
		sc.Scan()
		n2, _ := strconv.Atoi(sc.Text())

		// fmt.Println(n1, string(r[n1-1]), n2, string(r[n2-1]))

		if (r[n1-1] == pdca[0] && r[n2-1] == pdca[1]) ||
			(r[n1-1] == pdca[1] && r[n2-1] == pdca[2]) ||
			(r[n1-1] == pdca[2] && r[n2-1] == pdca[3]) {
			path[n1-1] = append(path[n1-1], n2-1)
		} else if (r[n2-1] == pdca[0] && r[n1-1] == pdca[1]) ||
			(r[n2-1] == pdca[1] && r[n1-1] == pdca[2]) ||
			(r[n2-1] == pdca[2] && r[n1-1] == pdca[3]) {
			path[n2-1] = append(path[n2-1], n1-1)
		}
	}

	// fmt.Println(path)

	// A, D, C, Pの順に、移動先が何通りになるかの評価
	// Aの移動先はないので、常に1通りになる
	// この時10^9 + 7を超える場合は余りを保持するようにする
	p := make([]int, n)
	ans := 0
	for i := len(pdca) - 1; i >= 0; i-- {
		for _, j := range sr[i] {
			if string(r[j]) == "A" {
				p[j] = 1
			} else {
				sum := 0
				for _, k := range path[j] {
					sum += p[k]
				}
				p[j] = sum % mod

				if i == 0 {
					ans += p[j]
				}
			}
		}
	}

	fmt.Println(ans % mod)
}

yukicoder

結果

テストケース

ソースコード