結果

問題 No.1266 7 Colors
ユーザー aru aru
提出日時 2020-12-01 22:35:02
言語 Go
(1.23.4)
結果
AC  
実行時間 199 ms / 3,000 ms
コード長 3,709 bytes
コンパイル時間 11,690 ms
コンパイル使用メモリ 233,848 KB
実行使用メモリ 26,272 KB
最終ジャッジ日時 2024-09-13 03:13:47
合計ジャッジ時間 17,724 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 19
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

package main
import (
    "bufio"
    "fmt"
    "os"
    "sort"
    "strconv"
)
var sc = bufio.NewScanner(os.Stdin)
var wr = bufio.NewWriter(os.Stdout)
func out(x ...interface{}) {
    fmt.Fprintln(wr, x...)
}
func getI() int {
    sc.Scan()
    i, e := strconv.Atoi(sc.Text())
    if e != nil {
        panic(e)
    }
    return i
}
func getF() float64 {
    sc.Scan()
    i, e := strconv.ParseFloat(sc.Text(), 64)
    if e != nil {
        panic(e)
    }
    return i
}
func getInts(N int) []int {
    ret := make([]int, N)
    for i := 0; i < N; i++ {
        ret[i] = getI()
    }
    return ret
}
func getS() string {
    sc.Scan()
    return sc.Text()
}
// min, max, asub, absなど基本関数
func max(a, b int) int {
    if a > b {
        return a
    }
    return b
}
func min(a, b int) int {
    if a < b {
        return a
    }
    return b
}
func asub(a, b int) int {
    if a > b {
        return a - b
    }
    return b - a
}
func abs(a int) int {
    if a >= 0 {
        return a
    }
    return -a
}
func lowerBound(a []int, x int) int {
    idx := sort.Search(len(a), func(i int) bool {
        return a[i] >= x
    })
    return idx
}
func upperBound(a []int, x int) int {
    idx := sort.Search(len(a), func(i int) bool {
        return a[i] > x
    })
    return idx
}
//
// Disjoint Set Union: Union Find Tree
//
// DSU :
type DSU struct {
    parentOrSize []int
    hasColor     []int
    n            int
}
// Dsu :
func Dsu(n int) *DSU {
    var d DSU
    d.n = n
    d.parentOrSize = make([]int, n)
    d.hasColor = make([]int, n)
    for i := 0; i < n; i++ {
        d.parentOrSize[i] = -1
    }
    return &d
}
// Merge :
func (d DSU) Merge(a, b int) int {
    x, y := d.Leader(a), d.Leader(b)
    if x == y {
        return x
    }
    if -d.parentOrSize[x] < -d.parentOrSize[y] {
        x, y = y, x
    }
    d.parentOrSize[x] += d.parentOrSize[y]
    d.hasColor[x] += d.hasColor[y]
    d.parentOrSize[y] = x
    return x
}
// Same :
func (d DSU) Same(a, b int) bool {
    return d.Leader(a) == d.Leader(b)
}
// Leader :
func (d DSU) Leader(a int) int {
    if d.parentOrSize[a] < 0 {
        return a
    }
    d.parentOrSize[a] = d.Leader(d.parentOrSize[a])
    return d.parentOrSize[a]
}
// SetColor :
func (d DSU) SetColor(a, c int) {
    x := d.Leader(a)
    d.hasColor[x] = c
}
// Size :
func (d DSU) Size(a int) int {
    return -d.parentOrSize[d.Leader(a)]
}
// CSize :
func (d DSU) CSize(a int) int {
    x := d.Leader(a)
    return d.hasColor[x]
}
// Groups : original implement
func (d DSU) Groups() [][]int {
    m := make(map[int][]int)
    for i := 0; i < d.n; i++ {
        x := d.Leader(i)
        if x < 0 {
            m[i] = append(m[i], i)
        } else {
            m[x] = append(m[x], i)
        }
    }
    ret := make([][]int, len(m))
    idx := 0
    for _, e := range m {
        ret[idx] = make([]int, len(e))
        copy(ret[idx], e)
        idx++
    }
    return ret
}
func main() {
    defer wr.Flush()
    sc.Split(bufio.ScanWords)
    sc.Buffer([]byte{}, 1000000)
    // this template is new version.
    // use getI(), getS(), getInts(), getF()
    N, M, Q := getI(), getI(), getI()
    uf := Dsu(N * 7)
    s := make([][]byte, N)
    for i := 0; i < N; i++ {
        s[i] = []byte(getS())
    }
    set := func(i int) {
        for j := 1; j < 7; j++ {
            if s[i][j] == '1' && s[i][j-1] == '1' {
                uf.Merge(i*7+j, i*7+j-1)
            }
            if s[i][0] == '1' && s[i][6] == '1' {
                uf.Merge(i*7, i*7+6)
            }
        }
    }
    for i := 0; i < N; i++ {
        set(i)
    }
    node := make([][]int, N)
    for i := 0; i < M; i++ {
        u, v := getI()-1, getI()-1
        node[u] = append(node[u], v)
        node[v] = append(node[v], u)
        for j := 0; j < 7; j++ {
            if s[u][j] == '1' && s[v][j] == '1' {
                uf.Merge(u*7+j, v*7+j)
            }
        }
    }
    for i := 0; i < Q; i++ {
        op, x, y := getI(), getI()-1, getI()-1
        if op == 2 {
            out(uf.Size(x * 7))
        } else {
            uf.SetColor(x*7+y, 1)
            s[x][y] = '1'
            set(x)
            for _, e := range node[x] {
                for j := 0; j < 7; j++ {
                    if s[x][j] == '1' && s[e][j] == '1' {
                        uf.Merge(x*7+j, e*7+j)
                    }
                }
            }
        }
    }
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0