結果

問題 No.1282 Display Elements
コンテスト
ユーザー ID 21712
提出日時 2026-05-29 13:10:38
言語 Go
(1.26.1)
コンパイル:
env GOCACHE=/tmp go build _filename_
実行:
./Main
結果
AC  
実行時間 160 ms / 2,000 ms
コード長 1,979 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 11,613 ms
コンパイル使用メモリ 280,552 KB
実行使用メモリ 15,616 KB
最終ジャッジ日時 2026-05-29 13:10:55
合計ジャッジ時間 14,838 ms
ジャッジサーバーID
（参考情報） 		judge2_0 / judge1_0
純コード判定待ち
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

package main

import . "fmt"
import . "os"
import bf "bufio"
import . "slices"

func main() {
	rd := bf.NewReader(Stdin)
	
	var n int
	Fscan(rd,&n)
	a := make([]int, n)
	for i := range a {
		Fscan(rd,&a[i])
	}
	b := make([]int, n)
	for i := range b {
		Fscan(rd,&b[i])
	}
	
	btree := newTree(Concat(a, b))
	
	Sort(a)
	
	ans := 0
	for i := 0; i < n; i++ {
		btree.add(b[i])
		ans += btree.lowerCount(a[i])
	}
	Println(ans)
}

/*
考察

累積和や統計の情報の取得や追加や更新が高速で行えるデータ構造やアルゴリズムを使って解く問題のように見える
aは昇順に並びかえて
BITや順序付きマップ(平衡二分探索木)などを使い
i番目の操作にb[i]を追加して、a[i]未満の個数を取得する、の繰り返し

Go言語の標準には実装されてないし
BITや平衡二分探索木をそらでは実装できないし
どちらも事前にコードを用意してないので（コンテストであれば不利）


*/


type Node struct {
	key, cnt, lc, uc int
	lower, upper *Node
}

func (n *Node)count() int {
	if n == nil {
		return 0
	}
	return n.cnt+n.lc+n.uc
}

func newTree(keys []int) *Node {
	keys = Clone(keys)
	if !IsSorted(keys) {
		Sort(keys)
	}
	keys = Compact(keys)
	return build(keys)
}

func build(keys []int) *Node {
	if len(keys) == 0 {
		return nil
	}
	n := new(Node)
	if len(keys) == 1 {
		n.key = keys[0]
		return n
	}
	p := len(keys)/2
	n.key= keys[p]
	n.lower = build(keys[:p])
	n.upper = build(keys[p+1:])
	return n
}

func (n *Node) add(key int) {
	switch {
		case n.key == key:
			n.cnt++
		case n.key < key:
			n.uc++
			n.upper.add(key)
		case n.key > key:
			n.lc++
			n.lower.add(key)
	}
}

func (n *Node)lowerCount(key int) int {
	switch {
		case n.key == key:
			return n.lc
		case n.key < key:
			return n.cnt+n.lc+n.upper.lowerCount(key)
		case n.key > key:
			return n.lower.lowerCount(key)
	}
	// 賢いコンパイラとかだと不要になるreturn
	return 0
}
0