結果
問題 | No.738 平らな農地 |
ユーザー | 草苺奶昔 |
提出日時 | 2024-04-25 13:53:41 |
言語 | Go (1.22.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 6,278 bytes |
コンパイル時間 | 14,136 ms |
コンパイル使用メモリ | 219,904 KB |
実行使用メモリ | 9,856 KB |
最終ジャッジ日時 | 2024-11-08 00:56:33 |
合計ジャッジ時間 | 20,642 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 1 ms
5,248 KB |
testcase_03 | WA | - |
testcase_04 | AC | 1 ms
5,248 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 | 3 ms
5,248 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 | 70 ms
5,504 KB |
testcase_66 | AC | 72 ms
6,656 KB |
testcase_67 | WA | - |
testcase_68 | AC | 93 ms
8,064 KB |
testcase_69 | WA | - |
testcase_70 | WA | - |
testcase_71 | AC | 34 ms
5,248 KB |
testcase_72 | AC | 53 ms
5,248 KB |
testcase_73 | AC | 55 ms
5,248 KB |
testcase_74 | AC | 61 ms
5,248 KB |
testcase_75 | WA | - |
testcase_76 | WA | - |
testcase_77 | WA | - |
testcase_78 | WA | - |
testcase_79 | WA | - |
testcase_80 | WA | - |
testcase_81 | AC | 102 ms
5,504 KB |
testcase_82 | AC | 96 ms
5,248 KB |
testcase_83 | AC | 53 ms
5,248 KB |
testcase_84 | AC | 58 ms
5,632 KB |
testcase_85 | WA | - |
testcase_86 | WA | - |
testcase_87 | AC | 93 ms
9,216 KB |
testcase_88 | AC | 90 ms
9,344 KB |
testcase_89 | AC | 1 ms
5,248 KB |
testcase_90 | AC | 1 ms
5,248 KB |
testcase_91 | AC | 1 ms
5,248 KB |
ソースコード
// 动态中位数,用两个可删除堆(对顶堆)实现 // api: // 1. Insert(x T) // 2. Discard(x T) bool // 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() 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.Println(M.Median()) // fmt.Println(M.DistToMedian(), 987, M.Size()) } fmt.Fprintln(out, res) } type S = int type DynamicMedian struct { size int32 lower *erasableHeap upper *erasableHeap lowerCounter map[S]int32 upperCounter map[S]int32 lowerSum S upperSum S } func NewDynamicMedian() *DynamicMedian { return &DynamicMedian{ lower: newErasableHeap(func(a, b S) bool { return a > b }, nil), upper: newErasableHeap(func(a, b S) bool { return a < b }, nil), lowerCounter: make(map[S]int32), upperCounter: make(map[S]int32), } } func (d *DynamicMedian) Insert(value S) { 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) Discard(value S) bool { if v1, ok1 := d.lowerCounter[value]; ok1 { d.lowerCounter[value]-- if v1 == 1 { delete(d.lowerCounter, value) } d.lower.Erase(value) d.lowerSum -= value d.size-- d.balance() return true } else if v2, ok2 := d.upperCounter[value]; ok2 { d.upperCounter[value]-- if v2 == 1 { delete(d.upperCounter, value) } d.upper.Erase(value) d.upperSum -= value d.size-- d.balance() return true } else { return false } } // 返回中位数.如果元素个数为偶数,返回两个中位数. func (d *DynamicMedian) Median() (low, high S) { 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) DistToMedian() S { if d.size == 0 { return 0 } low, _ := d.Median() sum1 := low*S(d.lower.Len()) - d.lowerSum if sum1 < 0 { sum1 = -sum1 } sum2 := d.upperSum - low*S(d.upper.Len()) if sum2 < 0 { sum2 = -sum2 } // fmt.Println(low, d.lowerSum, d.upperSum, sum1, sum2) return sum1 + sum2 } func (d *DynamicMedian) Size() int32 { return d.size } func (d *DynamicMedian) balance() { // 偶数个数时,|lower heap| == |upper heap| // 奇数个数时,|lower heap| + 1 == |upper heap| for d.lower.Len()+1 < d.upper.Len() { upperMin := d.upper.Pop() d.lower.Push(upperMin) d.lowerSum += upperMin d.upperSum -= upperMin } for d.lower.Len() > d.upper.Len() { lowerMin := d.lower.Pop() d.upper.Push(lowerMin) d.upperSum += lowerMin d.lowerSum -= lowerMin } if d.size&1 == 0 { if d.lower.Len() != d.upper.Len() { panic("size error") } } else { if d.lower.Len()+1 != d.upper.Len() { panic("size error") } } if d.lower.Len() == 0 || d.upper.Len() == 0 { return } if d.lower.Peek() > d.upper.Peek() { upperMin := d.upper.Pop() d.lower.Push(upperMin) d.lowerSum += upperMin d.upperSum -= upperMin lowerMax := d.lower.Pop() d.upper.Push(lowerMax) d.upperSum += lowerMax d.lowerSum -= lowerMax } } type erasableHeap struct { data *heap erased *heap size int32 } func newErasableHeap(less func(a, b S) bool, nums []S) *erasableHeap { return &erasableHeap{newHeap(less, nums), newHeap(less, nil), int32(len(nums))} } // 从堆中删除一个元素,要保证堆中存在该元素. func (h *erasableHeap) Erase(value S) { h.erased.Push(value) h.normalize() h.size-- } func (h *erasableHeap) Push(value S) { h.data.Push(value) h.normalize() h.size++ } func (h *erasableHeap) Pop() (value S) { value = h.data.Pop() h.normalize() h.size-- return } func (h *erasableHeap) Peek() (value S) { value = h.data.Top() return } func (h *erasableHeap) Len() int32 { return h.size } func (h *erasableHeap) Clear() { h.data.Clear() h.erased.Clear() h.size = 0 } func (h *erasableHeap) normalize() { for h.data.Len() > 0 && h.erased.Len() > 0 && h.data.Top() == h.erased.Top() { h.data.Pop() h.erased.Pop() } } func newHeap(less func(a, b S) bool, nums []S) *heap { nums = append(nums[:0:0], nums...) heap := &heap{less: less, data: nums} if len(nums) > 1 { heap.heapify() } return heap } type heap struct { data []S less func(a, b S) bool } func (h *heap) Push(value S) { h.data = append(h.data, value) h.pushUp(h.Len() - 1) } func (h *heap) Pop() (value S) { 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 *heap) Top() (value S) { if h.Len() == 0 { panic("heap is empty") } value = h.data[0] return } func (h *heap) Len() int { return len(h.data) } func (h *heap) Clear() { h.data = h.data[:0] } func (h *heap) heapify() { n := h.Len() for i := (n >> 1) - 1; i > -1; i-- { h.pushDown(i) } } func (h *heap) 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 *heap) 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 }