結果
問題 | No.1390 Get together |
ユーザー |
![]() |
提出日時 | 2021-02-19 20:40:10 |
言語 | Go (1.23.4) |
結果 |
AC
|
実行時間 | 139 ms / 2,000 ms |
コード長 | 3,326 bytes |
コンパイル時間 | 11,046 ms |
コンパイル使用メモリ | 229,556 KB |
実行使用メモリ | 29,616 KB |
最終ジャッジ日時 | 2024-09-16 16:36:45 |
合計ジャッジ時間 | 15,907 ms |
ジャッジサーバーID (参考情報) |
judge6 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 29 |
ソースコード
package mainimport ("bufio""fmt""math""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}// min for n entryfunc nmin(a ...int) int {ret := a[0]for _, e := range a {ret = min(ret, e)}return ret}// max for n entryfunc nmax(a ...int) int {ret := a[0]for _, e := range a {ret = max(ret, e)}return ret}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 []intn int}// newDsu :func newDsu(n int) *DSU {var d DSUd.n = nd.parentOrSize = 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.parentOrSize[y] = xreturn 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]}// Size :func (d DSU) Size(a int) int {return -d.parentOrSize[d.Leader(a)]}// Groups : original implementfunc (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 := 0for _, e := range m {ret[idx] = make([]int, len(e))copy(ret[idx], e)idx++}return ret}type pair struct {x, y int}func main() {defer wr.Flush()sc.Split(bufio.ScanWords)sc.Buffer([]byte{}, math.MaxInt32)// this template is new version.// use getI(), getS(), getInts(), getF()N, M := getI(), getI()box := make([][]pair, M)col := make([][]pair, N)for i := 0; i < N; i++ {b, c := getI()-1, getI()-1box[b] = append(box[b], pair{c, i})col[c] = append(col[c], pair{b, i})}uf := newDsu(N)for _, e := range box {if len(e) > 1 {p := e[0].yfor _, v := range e {uf.Merge(p, v.y)}}}ans := 0for _, e := range col {if len(e) > 1 {p := e[0].yfor _, v := range e {if uf.Same(p, v.y) == false {uf.Merge(p, v.y)ans++}}}}out(ans)}