[go刷题模板] SortedList/简单名次树

一、 算法&数据结构

1. 描述

py的SortedList超好用，但是代码很长，这里用简单的代码搞一个，但是不支持删除。

• 另外第k大也可以搞，有空再补。
• 不过实测同一道题2e5的数据量，go的SortedList性能比用树状数组差很多:1372ms -> 420ms

2. 复杂度分析

1. 查询query, O(log₂n)
2. 更新update，O(log₂n)

3. 常见应用

4. 常用优化

二、 模板代码

1. 查询v在数据中的名次/有多少数小于v/bisect_left(v)

• cf652d D. Nested Segments
  

package main

import (
	"bufio"
	. "fmt"
	"io"
	"os"
	"sort"
)

// SortedList 1372ms
func solve(in io.Reader, out io.Writer) {
	var n int
	Fscan(in, &n)
	type seg struct{ i, l, r int }
	a := make([]seg, n)
	for i := 0; i < n; i++ {
		Fscan(in, &a[i].l, &a[i].r)
		a[i].i = i
	}
	sort.Slice(a, func(i, j int) bool { return a[i].r < a[j].r })
	ans := make([]int, n)
	p := NewSortedList()
	for _, s := range a {
		l, i := s.l, s.i
		//ans[i] = len(p) - p.BisectLeft(l)
		ans[i] = p.Len() - p.BisectLeft(l)
		p.Add(l)
	}

	for _, v := range ans {
		Fprintln(out, v)
	}
}

func run(_r io.Reader, _w io.Writer) {
	in := bufio.NewReader(_r)
	out := bufio.NewWriter(_w)
	defer out.Flush()
	solve(in, out)
}

func main() { run(os.Stdin, os.Stdout) }

type SortedList struct {
	Small []int
	Large []int
}

func NewSortedList() SortedList {
	var small []int
	var large []int
	return SortedList{small, large}
}

func (u *SortedList) Add(v int) {
	if len(u.Small) > 6700 {
		u.Large = append(u.Large, u.Small...)
		u.Small = []int{}
		sort.Ints(u.Large)
	}
	if len(u.Small) == 0 || u.Small[len(u.Small)-1] <= v {
		u.Small = append(u.Small, v)
	} else if u.Small[0] >= v {
		u.Small = append([]int{v}, u.Small...)
	} else {
		p := sort.Search(len(u.Small), func(i int) bool { return u.Small[i] >= v })
		t := append([]int{}, u.Small[p:]...) // 实现insert 需要保存临时变量
		u.Small = append(u.Small[:p], v)
		u.Small = append(u.Small, t...)
		//u.Small = append(u.Small[:p], append([]int{v}, u.Small[p:]...)...)
	}
}

func (u SortedList) BisectLeft(v int) int {
	return sort.Search(len(u.Small), func(i int) bool { return u.Small[i] >= v }) + sort.Search(len(u.Large), func(i int) bool { return u.Large[i] >= v })
}

func (u SortedList) Len() int {
	return len(u.Small) + len(u.Large)
}

三、其他

四、更多例题

五、参考链接