#776289 (Go) No.1750 ラムドスウイルスの感染拡大-hard

• 問題: No.1750 ラムドスウイルスの感染拡大-hard
• ユーザー: Hima
• 提出日時: 2022-07-05 23:04:40
• 言語: Go (1.22.1)
• 結果: AC
• 実行時間: 1,643 ms / 2,000 ms
• コード長: 1,437 bytes
• コンパイル時間: 11,388 ms
• コンパイル使用メモリ: 225,016 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-05-09 19:13:15
• 合計ジャッジ時間: 29,594 ms
• ジャッジサーバーID （参考情報）: judge3 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	1 ms; 5,376 KB
testcase_04	AC	38 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	1 ms; 5,376 KB
testcase_08	AC	284 ms; 5,376 KB
testcase_09	AC	281 ms; 5,376 KB
testcase_10	AC	283 ms; 5,376 KB
testcase_11	AC	218 ms; 5,376 KB
testcase_12	AC	307 ms; 5,376 KB
testcase_13	AC	291 ms; 5,376 KB
testcase_14	AC	1,547 ms; 5,376 KB
testcase_15	AC	1,572 ms; 5,376 KB
testcase_16	AC	1,594 ms; 5,376 KB
testcase_17	AC	1,587 ms; 5,376 KB
testcase_18	AC	1,643 ms; 5,376 KB
testcase_19	AC	1,568 ms; 5,376 KB
testcase_20	AC	1,088 ms; 5,376 KB
testcase_21	AC	1,282 ms; 5,376 KB
testcase_22	AC	220 ms; 5,376 KB
testcase_23	AC	1,511 ms; 5,376 KB
testcase_24	AC	169 ms; 5,376 KB
testcase_25	AC	416 ms; 5,376 KB
testcase_26	AC	130 ms; 5,376 KB
testcase_27	AC	3 ms; 5,376 KB
testcase_28	AC	8 ms; 5,376 KB
testcase_29	AC	3 ms; 5,376 KB
testcase_30	AC	215 ms; 5,376 KB
testcase_31	AC	200 ms; 5,376 KB
testcase_32	AC	198 ms; 5,376 KB
testcase_33	AC	193 ms; 5,376 KB

ソースコード

package main

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

var sc = bufio.NewScanner(os.Stdin)
var out = bufio.NewWriter(os.Stdout)

func MulMatrix(n int, a, b [][]int, p int) [][]int {
	ret := make([][]int, n)
	for i := 0; i < n; i++ {
		ret[i] = make([]int, n)
	}
	for i := 0; i < n; i++ {
		for j := 0; j < n; j++ {
			for k := 0; k < n; k++ {
				ret[i][j] += a[i][k] * b[k][j]
				ret[i][j] %= p
			}
		}
	}
	return ret
}

func PowMatrix(n int, a [][]int, x, p int) [][]int {
	ret := make([][]int, n)
	for i := 0; i < n; i++ {
		ret[i] = make([]int, n)
	}
	for i := 0; i < n; i++ {
		ret[i][i] = 1
	}
	for x > 0 {
		if x%2 == 1 {
			ret = MulMatrix(n, ret, a, p)
		}
		x >>= 1
		a = MulMatrix(n, a, a, p)
	}
	return ret
}
func solve(n, m, lt int, s, t []int) int {
	const p = 998244353

	w := make([][]int, n)
	for i := 0; i < n; i++ {
		w[i] = make([]int, n)
	}
	for i := 0; i < m; i++ {
		w[s[i]][t[i]] = 1
		w[t[i]][s[i]] = 1
	}
	w = PowMatrix(n, w, lt, p)

	return w[0][0]
}

func main() {
	buf := make([]byte, 1024*1024)
	sc.Buffer(buf, bufio.MaxScanTokenSize)
	sc.Split(bufio.ScanWords)

	n, m, lt := nextInt(), nextInt(), nextInt()
	var s, t []int
	for i := 0; i < m; i++ {
		s = append(s, nextInt())
		t = append(t, nextInt())
	}

	ans := solve(n, m, lt, s, t)

	PrintInt(ans)
}

func nextInt() int {
	sc.Scan()
	i, _ := strconv.Atoi(sc.Text())
	return i
}

func PrintInt(x int) {
	defer out.Flush()
	fmt.Fprintln(out, x)
}