結果

問題 No.1477 Lamps on Graph
コンテスト
ユーザー ID 21712
提出日時 2026-05-24 03:25:41
言語 Go
(1.26.1)
コンパイル:
env GOCACHE=/tmp go build _filename_
実行:
./Main
結果
AC  
実行時間 237 ms / 2,000 ms
コード長 1,236 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 13,163 ms
コンパイル使用メモリ 287,464 KB
実行使用メモリ 15,104 KB
最終ジャッジ日時 2026-05-24 03:26:08
合計ジャッジ時間 22,071 ms
ジャッジサーバーID
（参考情報） 		judge1_0 / judge2_1
純コード判定待ち
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 38
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

package main

import . "fmt"
import . "os"
import bf "bufio"
import . "sort"

type T struct { id, value int }

func main() {
	rd := bf.NewReader(Stdin)
	wr := bf.NewWriter(Stdout)
	defer wr.Flush()
	
	var n,m int
	Fscan(rd,&n,&m)
	
	a := make([]int, n)
	for i := range a {
		Fscan(rd, &a[i])
	}
	
	graph := make([][]int, n)
	for i := 0; i < m; i++ {
		var u,v int
		Fscan(rd,&u,&v)
		u--
		v--
		if a[u] < a[v] {
			graph[u] = append(graph[u], v)
		}
		if a[u] > a[v] {
			graph[v] = append(graph[v], u)
		}
	}
	
	light := make([]bool, n)
	
	var k int
	Fscan(rd, &k)
	for i := 0; i < k; i++ {
		var b int
		Fscan(rd,&b)
		light[b-1] = true
	}
	
	ts := make([]*T, n)
	for i := range ts {
		ts[i] = &T{ 
			id: i, 
			value: a[i],
		}
	}
	Slice(ts, func(i, j int) bool {
		return ts[i].value < ts[j].value
	})

	ans := []int{}
	for _, t := range ts {
		if !light[t.id] {
			continue
		}
		ans = append(ans, t.id)
		for _, e := range graph[t.id] {
			light[e] = !light[e]
		}
	}
	
	Println(len(ans))
	for _, id := range ans {
		Println(id+1)
	}
}

/*
考察

因果的に
A[i]の小さい順に消灯していくだけだと思われる

グラフはA[u]<A[v]の方向の有向辺の有向グラフを構築しとくといいのかも


*/
0