結果
問題 | No.875 Range Mindex Query |
ユーザー |
|
提出日時 | 2023-06-14 23:43:55 |
言語 | TypeScript (5.7.2) |
結果 |
AC
|
実行時間 | 1,711 ms / 2,000 ms |
コード長 | 9,162 bytes |
コンパイル時間 | 9,441 ms |
コンパイル使用メモリ | 231,136 KB |
実行使用メモリ | 82,412 KB |
最終ジャッジ日時 | 2024-12-31 16:54:28 |
合計ジャッジ時間 | 23,369 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 1 |
other | AC * 18 |
ソースコード
import * as fs from 'fs'import { resolve } from 'path'type SegNode<E> = {left: SegNode<E> | undefinedright: SegNode<E> | undefinedindex: numberdata: Esum: E}class SegmentTreeDynamic<E = number> {private static _isPrimitive(o: unknown): o is number | string | boolean | symbol | bigint | null | undefined {return o === null || (typeof o !== 'object' && typeof o !== 'function')}private readonly _lower: numberprivate readonly _upper: numberprivate readonly _e: () => Eprivate readonly _op: (a: E, b: E) => Eprivate readonly _persistent: booleanprivate _root: SegNode<E>/*** 单点修改区间查询的动态开点懒标记线段树.线段树维护的值域为`[start, end)`.* @param start 值域下界.start>=0.* @param end 值域上界.* @param e 幺元.* @param op 结合律的二元操作.* @param persistent 是否持久化.持久化后,每次修改都会新建一个结点,否则会复用原来的结点.*/constructor(start: number, end: number, e: () => E, op: (a: E, b: E) => E, persistent = false) {if (persistent && !SegmentTreeDynamic._isPrimitive(e())) {throw new Error('persistent is only supported when e() return primitive values')}this._lower = startthis._upper = end + 5this._e = ethis._op = opthis._persistent = persistentthis._root = this.newRoot()}newRoot(): SegNode<E> {return undefined as any // nil}get(index: number, root: SegNode<E> = this._root): E {if (index < this._lower || index >= this._upper) return this._e()return this._get(root, index)}set(index: number, value: E, root: SegNode<E> = this._root): SegNode<E> {if (index < this._lower || index >= this._upper) return rootconst newRoot = this._set(root, this._lower, this._upper, index, value)this._root = newRootreturn newRoot}update(index: number, value: E, root: SegNode<E> = this._root): SegNode<E> {if (index < this._lower || index >= this._upper) return rootconst newRoot = this._update(root, this._lower, this._upper, index, value)this._root = newRootreturn newRoot}/*** 查询区间`[start,end)`的聚合值.* {@link _lower} <= start <= end <= {@link _upper}.*/query(start: number, end: number, root: SegNode<E> = this._root): E {if (start < this._lower) start = this._lowerif (end > this._upper) end = this._upperif (start >= end) return this._e()let res = this._e()const _query = (node: SegNode<E> | undefined, l: number, r: number, ql: number, qr: number) => {if (!node) returnql = l > ql ? l : qlqr = r < qr ? r : qrif (ql >= qr) returnif (l === ql && r === qr) {res = this._op(res, node.sum)return}const m = Math.floor(l + (r - l) / 2)_query(node.left, l, m, ql, qr)if (ql <= node.index && node.index < qr) {res = this._op(res, node.data)}_query(node.right, m, r, ql, qr)}_query(root, this._lower, this._upper, start, end)return res}queryAll(root: SegNode<E> = this._root): E {return root.sum}/*** 二分查询最大的`end`使得切片`[start:end)`内的聚合值满足`check`.* {@link _lower} <= start <= {@link _upper}.* @alias findFirst*/maxRight(start: number, check: (e: E) => boolean, root: SegNode<E> = this._root): number {if (start < this._lower) start = this._lowerif (start >= this._upper) return this._upperlet x = this._e()const _maxRight = (node: SegNode<E> | undefined, l: number, r: number, ql: number): number => {if (!node || r <= ql) return this._upperconst tmp = this._op(x, node.sum)if (check(tmp)) {x = tmpreturn this._upper}const m = Math.floor(l + (r - l) / 2)const k = _maxRight(node.left, l, m, ql)if (k !== this._upper) return kif (ql <= node.index) {x = this._op(x, node.data)if (!check(x)) {return node.index}}return _maxRight(node.right, m, r, ql)}return _maxRight(root, this._lower, this._upper, start)}/*** 二分查询最小的`start`使得切片`[start:end)`内的聚合值满足`check`.* {@link _lower} <= end <= {@link _upper}.* @alias findLast*/minLeft(end: number, check: (e: E) => boolean, root: SegNode<E> = this._root): number {if (end > this._upper) end = this._upperif (end <= this._lower) return this._lowerlet x = this._e()const _minLeft = (node: SegNode<E> | undefined, l: number, r: number, qr: number): number => {if (!node || qr <= l) return this._lowerconst tmp = this._op(node.sum, x)if (check(tmp)) {x = tmpreturn this._lower}const m = Math.floor(l + (r - l) / 2)const k = _minLeft(node.right, m, r, qr)if (k !== this._lower) return kif (node.index < qr) {x = this._op(node.data, x)if (!check(x)) {return node.index + 1}}return _minLeft(node.left, l, m, qr)}return _minLeft(root, this._lower, this._upper, end)}getAll(root: SegNode<E> = this._root): [index: number, value: E][] {const res: [number, E][] = []const _getAll = (node: SegNode<E> | undefined) => {if (!node) return_getAll(node.left)res.push([node.index, node.data])_getAll(node.right)}_getAll(root)return res}private _copyNode(node: SegNode<E>): SegNode<E> {if (!node || !this._persistent) return nodereturn { left: node.left, right: node.right, index: node.index, data: node.data, sum: node.sum }}private _get(root: SegNode<E> | undefined, index: number): E {if (!root) return this._e()if (index === root.index) return root.dataif (index < root.index) return this._get(root.left, index)return this._get(root.right, index)}private _set(root: SegNode<E> | undefined, l: number, r: number, i: number, x: E): SegNode<E> {if (!root) return SegmentTreeDynamic._newNode(i, x)root = this._copyNode(root)if (root.index === i) {root.data = xthis._pushUp(root)return root}const m = Math.floor(l + (r - l) / 2)if (i < m) {if (root.index < i) {const tmp1 = root.indexroot.index = ii = tmp1const tmp2 = root.dataroot.data = xx = tmp2}root.left = this._set(root.left, l, m, i, x)} else {if (i < root.index) {const tmp1 = root.indexroot.index = ii = tmp1const tmp2 = root.dataroot.data = xx = tmp2}root.right = this._set(root.right, m, r, i, x)}this._pushUp(root)return root}private _pushUp(root: SegNode<E>): void {root.sum = root.dataif (root.left) root.sum = this._op(root.left.sum, root.sum)if (root.right) root.sum = this._op(root.sum, root.right.sum)}private _update(root: SegNode<E> | undefined, l: number, r: number, i: number, x: E): SegNode<E> {if (!root) return SegmentTreeDynamic._newNode(i, x)root = this._copyNode(root)if (root.index === i) {root.data = this._op(root.data, x)this._pushUp(root)return root}const m = Math.floor(l + (r - l) / 2)if (i < m) {if (root.index < i) {const tmp1 = root.indexroot.index = ii = tmp1const tmp2 = root.dataroot.data = xx = tmp2}root.left = this._update(root.left, l, m, i, x)} else {if (i < root.index) {const tmp1 = root.indexroot.index = ii = tmp1const tmp2 = root.dataroot.data = xx = tmp2}root.right = this._update(root.right, m, r, i, x)}this._pushUp(root)return root}private static _newNode<V>(index: number, value: V): SegNode<V> {return {index,left: undefined,right: undefined,data: value,sum: value}}}function useInput(path?: string) {let data: stringif (path) {data = fs.readFileSync(resolve(__dirname, path), 'utf8')} else {data = fs.readFileSync(process.stdin.fd, 'utf8')}const lines = data.split(/\r\n|\r|\n/)let lineId = 0const input = (): string => lines[lineId++]return {input}}const { input } = useInput()const [N, Q] = input().split(' ').map(Number)const a = input().split(' ').map(Number)const INF = 2e15const seg = new SegmentTreeDynamic(0, a.length, () => INF, Math.min,true)a.forEach((v, i) => seg.set(i, v))for (let i = 0; i < Q; i++) {let [c, l, r] = input().split(' ').map(Number)if (c === 1) {l--r--const vl = seg.get(l)seg.set(l, seg.get(r))seg.set(r, vl)} else {l--const mn = seg.query(l, r)const a1 = seg.maxRight(l, n => n > mn)let a2 = seg.minLeft(r, n => n > mn)a2--if (a1 !== a2) {throw new Error('a1 !== a2')}console.log(a1 + 1)}}