#856121 (Go) No.1320 Two Type Min Cost Cycle

提出ソース

結果

問題	No.1320 Two Type Min Cost Cycle
ユーザー	草苺奶昔
提出日時	2023-04-02 19:25:25
言語	Go (1.22.1)
結果	AC
実行時間	213 ms / 2,000 ms
コード長	5,215 bytes
コンパイル時間	16,005 ms
コンパイル使用メモリ	219,128 KB
実行使用メモリ	8,360 KB
最終ジャッジ日時	2023-10-25 05:10:41
合計ジャッジ時間	15,688 ms
ジャッジサーバーID （参考情報）	judge15 / judge11

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,348 KB
testcase_01	AC	1 ms 4,348 KB
testcase_02	AC	1 ms 4,348 KB
testcase_03	AC	1 ms 4,348 KB
testcase_04	AC	2 ms 4,348 KB
testcase_05	AC	2 ms 4,348 KB
testcase_06	AC	1 ms 4,348 KB
testcase_07	AC	6 ms 8,112 KB
testcase_08	AC	6 ms 8,060 KB
testcase_09	AC	143 ms 8,248 KB
testcase_10	AC	8 ms 8,176 KB
testcase_11	AC	124 ms 8,312 KB
testcase_12	AC	16 ms 8,248 KB
testcase_13	AC	57 ms 8,284 KB
testcase_14	AC	4 ms 7,856 KB
testcase_15	AC	8 ms 8,172 KB
testcase_16	AC	13 ms 5,692 KB
testcase_17	AC	4 ms 7,848 KB
testcase_18	AC	5 ms 7,864 KB
testcase_19	AC	66 ms 8,300 KB
testcase_20	AC	2 ms 5,620 KB
testcase_21	AC	136 ms 8,308 KB
testcase_22	AC	1 ms 4,348 KB
testcase_23	AC	1 ms 4,348 KB
testcase_24	AC	1 ms 4,348 KB
testcase_25	AC	1 ms 4,348 KB
testcase_26	AC	1 ms 4,348 KB
testcase_27	AC	1 ms 4,348 KB
testcase_28	AC	4 ms 7,836 KB
testcase_29	AC	68 ms 8,168 KB
testcase_30	AC	1 ms 4,348 KB
testcase_31	AC	13 ms 7,832 KB
testcase_32	AC	2 ms 4,348 KB
testcase_33	AC	74 ms 8,172 KB
testcase_34	AC	35 ms 8,112 KB
testcase_35	AC	7 ms 5,692 KB
testcase_36	AC	5 ms 4,348 KB
testcase_37	AC	2 ms 4,348 KB
testcase_38	AC	10 ms 5,712 KB
testcase_39	AC	3 ms 4,348 KB
testcase_40	AC	2 ms 4,348 KB
testcase_41	AC	1 ms 4,348 KB
testcase_42	AC	1 ms 4,348 KB
testcase_43	AC	12 ms 8,176 KB
testcase_44	AC	8 ms 8,176 KB
testcase_45	AC	213 ms 8,240 KB
testcase_46	AC	11 ms 7,912 KB
testcase_47	AC	34 ms 8,156 KB
testcase_48	AC	61 ms 8,148 KB
testcase_49	AC	8 ms 5,628 KB
testcase_50	AC	1 ms 4,348 KB
testcase_51	AC	1 ms 4,348 KB
testcase_52	AC	15 ms 7,740 KB
testcase_53	AC	6 ms 5,652 KB
testcase_54	AC	199 ms 8,356 KB
testcase_55	AC	197 ms 8,356 KB
testcase_56	AC	199 ms 8,356 KB
testcase_57	AC	188 ms 8,356 KB
testcase_58	AC	193 ms 8,360 KB
testcase_59	AC	198 ms 8,360 KB

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

ソースコード

raw source code

package main

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

func main() {
	in := bufio.NewReader(os.Stdin)
	out := bufio.NewWriter(os.Stdout)
	defer out.Flush()

	var directed, n, m int
	fmt.Fscan(in, &directed, &n, &m)
	edges := make([]Edge, 0, m)
	for i := 0; i < m; i++ {
		var u, v, w int
		fmt.Fscan(in, &u, &v, &w)
		u--
		v--
		edges = append(edges, Edge{u, v, w})
	}
	res := MincostCycle(n, edges, directed == 1)
	if res == INF {
		res = -1
	}
	fmt.Fprintln(out, res)
}

const INF int = 1e18

type Edge struct{ from, to, weight int }

// 返回最小环的权值和. 不存在时返回INF.
//  !把每条边断开,然后求从断开的边的两个端点到另一端点的最短路.
func MincostCycle(n int, edges []Edge, directed bool) int {
	adjList := make([][][2]int, n)
	maxWeight := 0
	for _, e := range edges {
		u, v, w := e.from, e.to, e.weight
		if w > maxWeight {
			maxWeight = w
		}
		adjList[u] = append(adjList[u], [2]int{v, w})
		if !directed {
			adjList[v] = append(adjList[v], [2]int{u, w})
		}
	}
	res := INF
	for _, e := range edges {
		from_, to, weight := e.from, e.to, e.weight
		var dist int
		if maxWeight <= 1 {
			dist = bfs01Point(n, adjList, to, from_)
		} else {
			dist = dijkstraPoint(n, adjList, to, from_)
		}
		cand := weight + dist
		if cand < res {
			res = cand
		}
	}
	return res
}

func min(a, b int) int {
	if a < b {
		return a
	}
	return b
}

func bfs01Point(n int, adjList [][][2]int, start, target int) int {
	dist := make([]int, n)
	for i := range dist {
		dist[i] = INF
	}

	queue := Deque{}
	queue.Append(start)
	dist[start] = 0
	for !queue.Empty() {
		cur := queue.PopLeft()
		if cur == target {
			return dist[cur]
		}
		for _, e := range adjList[cur] {
			next, cost := e[0], e[1]
			if (next == start && cur == target) || (next == target && cur == start) {
				continue
			}
			if cand := dist[cur] + cost; dist[next] > cand {
				dist[next] = cand
				if cost == 0 {
					queue.AppendLeft(next)
				} else {
					queue.Append(next)
				}
			}
		}
	}

	return INF
}

type D = int
type Deque struct{ l, r []D }

func (q Deque) Empty() bool {
	return len(q.l) == 0 && len(q.r) == 0
}

func (q Deque) Size() int {
	return len(q.l) + len(q.r)
}

func (q *Deque) AppendLeft(v D) {
	q.l = append(q.l, v)
}

func (q *Deque) Append(v D) {
	q.r = append(q.r, v)
}

func (q *Deque) PopLeft() (v D) {
	if len(q.l) > 0 {
		q.l, v = q.l[:len(q.l)-1], q.l[len(q.l)-1]
	} else {
		v, q.r = q.r[0], q.r[1:]
	}
	return
}

func (q *Deque) Pop() (v D) {
	if len(q.r) > 0 {
		q.r, v = q.r[:len(q.r)-1], q.r[len(q.r)-1]
	} else {
		v, q.l = q.l[0], q.l[1:]
	}
	return
}

func (q Deque) Front() D {
	if len(q.l) > 0 {
		return q.l[len(q.l)-1]
	}
	return q.r[0]
}

func (q Deque) Back() D {
	if len(q.r) > 0 {
		return q.r[len(q.r)-1]
	}
	return q.l[0]
}

// 0 <= i < q.Size()
func (q Deque) At(i int) D {
	if i < len(q.l) {
		return q.l[len(q.l)-1-i]
	}
	return q.r[i-len(q.l)]
}

func dijkstraPoint(n int, adjList [][][2]int, start, target int) int {
	dist := make([]int, n)
	for i := range dist {
		dist[i] = INF
	}
	dist[start] = 0

	pq := nhp(func(a, b H) int {
		return a.dist - b.dist
	}, []H{{start, 0}})

	for pq.Len() > 0 {
		curNode := pq.Pop()
		cur, curDist := curNode.node, curNode.dist
		if cur == target {
			return curDist
		}
		if curDist > dist[cur] {
			continue
		}

		for _, edge := range adjList[cur] {
			next, weight := edge[0], edge[1]
			if (next == start && cur == target) || (next == target && cur == start) {
				continue
			}
			if cand := curDist + weight; cand < dist[next] {
				dist[next] = cand
				pq.Push(H{next, cand})
			}
		}
	}

	return INF
}

type H = struct{ node, dist int }

// Should return a number:
//    negative , if a < b
//    zero     , if a == b
//    positive , if a > b
type Comparator func(a, b H) int

func nhp(comparator Comparator, nums []H) *Heap {
	nums = append(nums[:0:0], nums...)
	heap := &Heap{comparator: comparator, data: nums}
	heap.heapify()
	return heap
}

type Heap struct {
	data       []H
	comparator Comparator
}

func (h *Heap) Push(value H) {
	h.data = append(h.data, value)
	h.pushUp(h.Len() - 1)
}

func (h *Heap) Pop() (value H) {
	if h.Len() == 0 {
		return
	}

	value = h.data[0]
	h.data[0] = h.data[h.Len()-1]
	h.data = h.data[:h.Len()-1]
	h.pushDown(0)
	return
}

func (h *Heap) Peek() (value H) {
	if h.Len() == 0 {
		return
	}
	value = h.data[0]
	return
}

func (h *Heap) Len() int { return len(h.data) }

func (h *Heap) heapify() {
	n := h.Len()
	for i := (n >> 1) - 1; i > -1; i-- {
		h.pushDown(i)
	}
}

func (h *Heap) pushUp(root int) {
	for parent := (root - 1) >> 1; parent >= 0 && h.comparator(h.data[root], h.data[parent]) < 0; parent = (root - 1) >> 1 {
		h.data[root], h.data[parent] = h.data[parent], h.data[root]
		root = parent
	}
}

func (h *Heap) pushDown(root int) {
	n := h.Len()
	for left := (root<<1 + 1); left < n; left = (root<<1 + 1) {
		right := left + 1
		minIndex := root

		if h.comparator(h.data[left], h.data[minIndex]) < 0 {
			minIndex = left
		}

		if right < n && h.comparator(h.data[right], h.data[minIndex]) < 0 {
			minIndex = right
		}

		if minIndex == root {
			return
		}

		h.data[root], h.data[minIndex] = h.data[minIndex], h.data[root]
		root = minIndex
	}
}

yukicoder

結果

テストケース

ソースコード