結果

問題 No.1705 Mode of long array
ユーザー 小野寺健小野寺健
提出日時 2021-12-17 11:29:10
言語 Go
(1.22.1)
結果
WA  
実行時間 -
コード長 1,534 bytes
コンパイル時間 12,138 ms
コンパイル使用メモリ 236,220 KB
実行使用メモリ 7,680 KB
最終ジャッジ日時 2024-09-14 12:12:21
合計ジャッジ時間 19,916 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
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 WA -
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 -
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import (
	"bufio"
	"os"
	"strconv"
)

type KeyValue struct {
	key int
	value int
}

type SegmentTree struct {
	n int
	dat []*KeyValue
}

func (this *SegmentTree) new(n int) {
	this.n = 1
	for this.n < n {
		this.n *= 2
	}
	this.dat = make([]*KeyValue, 2*this.n-1)
	for i := 0; i < 2*this.n-1; i++ {
		this.dat[i] = &KeyValue{-1, -1}
	}
}

func (this *SegmentTree) update(k int, a int) {
	i := k
	k += this.n - 1
	this.dat[k].key = i
	this.dat[k].value = a
	for k > 0 {
		k = (k - 1) / 2
		this.dat[k] = Max(this.dat[k * 2 + 1], this.dat[k * 2 + 2])
	}
}

func (this SegmentTree) max() int {
	return this.dat[0].key + 1
}

func (this SegmentTree) get(i int) int {
	return this.dat[i + this.n - 1].value
}

func Max(a *KeyValue, b *KeyValue) *KeyValue{
	if a.value > b.value {
		return a
	} else if a.value < b.value {
		return b
	} else if a.key >= b.key {
		return a
	} else {
		return b
	}
}

func main() {
	sc := bufio.NewScanner(os.Stdin)
	sc.Split(bufio.ScanWords)
	sc.Scan()
	sc.Scan()
	M, _ := strconv.Atoi(sc.Text())	
	var tree SegmentTree
	tree.new(M)
	for i := 0; i < M; i++ {
		sc.Scan()
		a, _ := strconv.Atoi(sc.Text())
		tree.update(i, a)
	}

	sc.Scan()
	Q, _ := strconv.Atoi(sc.Text())	
	for i := 0; i < Q; i++ {
		sc.Scan()
		t, _ := strconv.Atoi(sc.Text())
		sc.Scan()
		x, _ := strconv.Atoi(sc.Text())
		sc.Scan()
		y, _ := strconv.Atoi(sc.Text())
		if t == 1 {
			tree.update(x-1, tree.get(x-1) + y)
		} else if t == 2 {
			tree.update(x-1, tree.get(x-1) - y)
		} else {
			println(tree.max())
		}
	}
}
0