結果
問題 | No.92 逃走経路 |
ユーザー |
![]() |
提出日時 | 2020-06-02 15:59:28 |
言語 | Go (1.23.4) |
結果 |
AC
|
実行時間 | 26 ms / 5,000 ms |
コード長 | 2,285 bytes |
コンパイル時間 | 14,170 ms |
コンパイル使用メモリ | 231,020 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-11-24 01:23:51 |
合計ジャッジ時間 | 15,227 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 18 |
ソースコード
package mainimport ("bufio""fmt""math/big""os""strconv""strings")func getScanner(fp *os.File) *bufio.Scanner {scanner := bufio.NewScanner(fp)scanner.Split(bufio.ScanWords)scanner.Buffer(make([]byte, 1000005), 1000005)return scanner}func getNextString(scanner *bufio.Scanner) string {scanner.Scan()return scanner.Text()}func getNextInt(scanner *bufio.Scanner) int {i, _ := strconv.Atoi(getNextString(scanner))return i}func getNextInt64(scanner *bufio.Scanner) int64 {i, _ := strconv.ParseInt(getNextString(scanner), 10, 64)return i}func getNextUint64(scanner *bufio.Scanner) uint64 {i, _ := strconv.ParseUint(getNextString(scanner), 10, 64)return i}func getNextFloat64(scanner *bufio.Scanner) float64 {i, _ := strconv.ParseFloat(getNextString(scanner), 64)return i}func main() {fp := os.Stdinwfp := os.Stdoutcnt := 0if os.Getenv("MASPY") == "ますピ" {fp, _ = os.Open(os.Getenv("BEET_THE_HARMONY_OF_PERFECT"))cnt = 1}if os.Getenv("MASPYPY") == "ますピッピ" {wfp, _ = os.Create(os.Getenv("NGTKANA_IS_GENIUS10"))}scanner := getScanner(fp)writer := bufio.NewWriter(wfp)solve(scanner, writer)for i := 0; i < cnt; i++ {fmt.Fprintln(writer, "-----------------------------------")solve(scanner, writer)}writer.Flush()}func solve(scanner *bufio.Scanner, writer *bufio.Writer) {n := getNextInt(scanner)m := getNextInt(scanner)k := getNextInt(scanner)pp := map[int][]pair{}for i := 0; i < m; i++ {a := getNextInt(scanner) - 1b := getNextInt(scanner) - 1c := getNextInt(scanner)pp[c] = append(pp[c], newPair(a, b))}dp := make([]*big.Int, k+1)for i := 0; i < k+1; i++ {dp[i] = big.NewInt(0)}for i := 0; i < n; i++ {dp[0].SetBit(dp[0], i, 1)}for i := 0; i < k; i++ {d := getNextInt(scanner)for _, p := range pp[d] {if dp[i].Bit(p[0]) == 1 {dp[i+1].SetBit(dp[i+1], p[1], 1)}if dp[i].Bit(p[1]) == 1 {dp[i+1].SetBit(dp[i+1], p[0], 1)}}}ans := make([]string, 0)for i := 0; i < n; i++ {if dp[k].Bit(i) == 1 {ans = append(ans, fmt.Sprintf("%d", i+1))}}fmt.Fprintln(writer, len(ans))fmt.Fprintln(writer, strings.Join(ans, " "))}func newPair(a, b int) pair {return pair{a, b}}type pair [2]int