結果

問題 No.738 平らな農地
ユーザー 草苺奶昔草苺奶昔
提出日時 2024-04-25 00:43:40
言語 Go
(1.22.1)
結果
WA  
実行時間 -
コード長 6,109 bytes
コンパイル時間 17,288 ms
コンパイル使用メモリ 225,244 KB
実行使用メモリ 12,544 KB
最終ジャッジ日時 2024-04-25 00:44:07
合計ジャッジ時間 21,754 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 WA -
testcase_04 AC 0 ms
5,376 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 AC 2 ms
5,376 KB
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 WA -
testcase_41 WA -
testcase_42 WA -
testcase_43 WA -
testcase_44 WA -
testcase_45 WA -
testcase_46 WA -
testcase_47 WA -
testcase_48 WA -
testcase_49 WA -
testcase_50 WA -
testcase_51 WA -
testcase_52 WA -
testcase_53 WA -
testcase_54 WA -
testcase_55 WA -
testcase_56 WA -
testcase_57 WA -
testcase_58 WA -
testcase_59 WA -
testcase_60 WA -
testcase_61 WA -
testcase_62 WA -
testcase_63 WA -
testcase_64 WA -
testcase_65 AC 87 ms
10,880 KB
testcase_66 AC 90 ms
11,648 KB
testcase_67 WA -
testcase_68 AC 86 ms
10,240 KB
testcase_69 WA -
testcase_70 WA -
testcase_71 AC 32 ms
5,376 KB
testcase_72 AC 49 ms
5,376 KB
testcase_73 AC 58 ms
5,376 KB
testcase_74 AC 58 ms
5,376 KB
testcase_75 WA -
testcase_76 WA -
testcase_77 WA -
testcase_78 WA -
testcase_79 WA -
testcase_80 WA -
testcase_81 WA -
testcase_82 AC 96 ms
8,320 KB
testcase_83 AC 48 ms
5,376 KB
testcase_84 WA -
testcase_85 WA -
testcase_86 WA -
testcase_87 AC 83 ms
8,960 KB
testcase_88 AC 81 ms
9,600 KB
testcase_89 AC 1 ms
5,376 KB
testcase_90 AC 1 ms
5,376 KB
testcase_91 AC 1 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

// 动态中位数,用两个可删除堆(对顶堆)实现
// api:
// 1. Insert(x T)
// 2. Erase(x T)
// 3. Median() (low, high T)
// 4. DistToMedian() T
// 5. Size() int32

package main

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

func main() {
	yuki738()
}

const INF int = 1e18

// No.738 平らな農地
// https://yukicoder.me/problems/no/738
// !滑动窗口所有数到中位数的距离和
func yuki738() {
	in := bufio.NewReader(os.Stdin)
	out := bufio.NewWriter(os.Stdout)
	defer out.Flush()

	var n, k int32
	fmt.Fscan(in, &n, &k)
	nums := make([]int, n)
	for i := int32(0); i < n; i++ {
		fmt.Fscan(in, &nums[i])
	}

	M := NewDynamicMedian[int]()
	res := INF

	for i := int32(0); i < n; i++ {
		M.Insert(nums[i])
		if i >= k {
			M.Discard(nums[i-k])
		}
		if i >= k-1 {
			res = min(res, M.DistToMedian())
		}
	}
	fmt.Fprintln(out, res)
}

type Integer interface {
	int | uint | int8 | uint8 | int16 | uint16 | int32 | uint32 | int64 | uint64
}

type DynamicMedian[V Integer] struct {
	size         int32
	lower        *erasableHeapGeneric[V]
	upper        *erasableHeapGeneric[V]
	lowerSum     V
	upperSum     V
	lowerCounter map[V]int32
	upperCounter map[V]int32
}

func NewDynamicMedian[V Integer]() *DynamicMedian[V] {
	return &DynamicMedian[V]{
		lower:        newErasableHeapGeneric[V](func(a, b V) bool { return a > b }),
		upper:        newErasableHeapGeneric[V](func(a, b V) bool { return a < b }),
		lowerCounter: make(map[V]int32),
		upperCounter: make(map[V]int32),
	}
}

func (d *DynamicMedian[V]) Insert(value V) {
	if d.size&1 == 0 {
		d.upper.Push(value)
		d.upperSum += value
		d.upperCounter[value]++
	} else {
		d.lower.Push(value)
		d.lowerSum += value
		d.lowerCounter[value]++
	}
	d.size++
	d.balance()
}

func (d *DynamicMedian[V]) Discard(value V) bool {
	if d.lowerCounter[value] > 0 {
		d.lowerCounter[value]--
		d.lower.Erase(value)
		d.lowerSum -= value
		d.size--
		return true
	} else if d.upperCounter[value] > 0 {
		d.upperCounter[value]--
		d.upper.Erase(value)
		d.upperSum -= value
		d.size--
		return true
	} else {
		return false
	}
}

// 返回中位数.如果元素个数为偶数,返回两个中位数.
func (d *DynamicMedian[V]) Median() (low, high V) {
	if d.size == 0 {
		return
	}
	if d.size&1 == 0 {
		low = d.lower.Peek()
		high = d.upper.Peek()
	} else {
		low = d.upper.Peek()
		high = low
	}
	return
}

func (d *DynamicMedian[V]) DistToMedian() V {
	if d.size == 0 {
		return 0
	}
	low, _ := d.Median()
	sum1 := low*V(d.lower.Len()) - d.lowerSum
	if sum1 < 0 {
		sum1 = -sum1
	}
	sum2 := low*V(d.upper.Len()) - d.upperSum
	if sum2 < 0 {
		sum2 = -sum2
	}
	return sum1 + sum2
}

func (d *DynamicMedian[V]) Size() int32 { return d.size }

func (d *DynamicMedian[V]) balance() {
	// 偶数个数时,|lower heap| == |upper heap|
	// 奇数个数时,|lower heap| + 1 == |upper heap|
	for d.lower.Len()+1 < d.upper.Len() {
		d.lower.Push(d.upper.Pop())
	}
	for d.lower.Len() > d.upper.Len() {
		d.upper.Push(d.lower.Pop())
	}
	if d.lower.Len() == 0 || d.upper.Len() == 0 {
		return
	}
	if v1, v2 := d.lower.Peek(), d.upper.Peek(); v1 > v2 {
		upperValue := d.upper.Pop()
		d.lower.Push(upperValue)
		d.lowerSum += upperValue
		d.upperSum -= upperValue

		lowerValue := d.lower.Pop()
		d.upper.Push(lowerValue)
		d.upperSum += lowerValue
		d.lowerSum -= lowerValue
	}
}

type erasableHeapGeneric[V comparable] struct {
	data   *heapGeneric[V]
	erased *heapGeneric[V]
}

func newErasableHeapGeneric[V comparable](less func(a, b V) bool, nums ...V) *erasableHeapGeneric[V] {
	return &erasableHeapGeneric[V]{newHeapGeneric(less, nums...), newHeapGeneric(less)}
}

// 从堆中删除一个元素,要保证堆中存在该元素.
func (h *erasableHeapGeneric[V]) Erase(value V) {
	h.erased.Push(value)
	h.normalize()
}

func (h *erasableHeapGeneric[V]) Push(value V) {
	h.data.Push(value)
	h.normalize()
}

func (h *erasableHeapGeneric[V]) Pop() (value V) {
	value = h.data.Pop()
	h.normalize()
	return
}

func (h *erasableHeapGeneric[V]) Peek() (value V) {
	value = h.data.Top()
	return
}

func (h *erasableHeapGeneric[V]) Len() int {
	return h.data.Len()
}

func (h *erasableHeapGeneric[V]) Clear() {
	h.data.Clear()
	h.erased.Clear()
}

func (h *erasableHeapGeneric[V]) normalize() {
	for h.data.Len() > 0 && h.erased.Len() > 0 && h.data.Top() == h.erased.Top() {
		h.data.Pop()
		h.erased.Pop()
	}
}

type heapGeneric[V comparable] struct {
	data []V
	less func(a, b V) bool
}

func newHeapGeneric[V comparable](less func(a, b V) bool, nums ...V) *heapGeneric[V] {
	nums = append(nums[:0:0], nums...)
	heap := &heapGeneric[V]{less: less, data: nums}
	if len(nums) > 1 {
		heap.heapify()
	}
	return heap
}

func (h *heapGeneric[V]) Push(value V) {
	h.data = append(h.data, value)
	h.pushUp(h.Len() - 1)
}

func (h *heapGeneric[V]) Pop() (value V) {
	if h.Len() == 0 {
		panic("heap is empty")
	}

	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 *heapGeneric[V]) Top() (value V) {
	if h.Len() == 0 {
		panic("heap is empty")
	}
	value = h.data[0]
	return
}

func (h *heapGeneric[V]) Len() int { return len(h.data) }

func (h *heapGeneric[V]) Clear() {
	h.data = h.data[:0]
}

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

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

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

		if h.less(h.data[left], h.data[minIndex]) {
			minIndex = left
		}

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

		if minIndex == root {
			return
		}

		h.data[root], h.data[minIndex] = h.data[minIndex], h.data[root]
		root = minIndex
	}
}
func min(a, b int) int {
	if a < b {
		return a
	}
	return b
}

func max(a, b int) int {
	if a > b {
		return a
	}
	return b
}
0