結果
問題 | No.693 square1001 and Permutation 2 |
ユーザー |
![]() |
提出日時 | 2020-10-11 16:09:58 |
言語 | Go (1.23.4) |
結果 |
AC
|
実行時間 | 2 ms / 2,000 ms |
コード長 | 2,197 bytes |
コンパイル時間 | 12,139 ms |
コンパイル使用メモリ | 233,200 KB |
実行使用メモリ | 5,376 KB |
最終ジャッジ日時 | 2024-07-20 17:25:34 |
合計ジャッジ時間 | 12,902 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 9 |
ソースコード
package mainimport ("bufio""container/heap""fmt""os""sort""strconv")var sc = bufio.NewScanner(os.Stdin)var wr = bufio.NewWriter(os.Stdout)func out(x ...interface{}) {fmt.Fprintln(wr, x...)}func getI() int {sc.Scan()i, e := strconv.Atoi(sc.Text())if e != nil {panic(e)}return i}func getF() float64 {sc.Scan()i, e := strconv.ParseFloat(sc.Text(), 64)if e != nil {panic(e)}return i}func getInts(N int) []int {ret := make([]int, N)for i := 0; i < N; i++ {ret[i] = getI()}return ret}func getS() string {sc.Scan()return sc.Text()}// min, max, asub, absなど基本関数func max(a, b int) int {if a > b {return a}return b}func min(a, b int) int {if a < b {return a}return b}func asub(a, b int) int {if a > b {return a - b}return b - a}func abs(a int) int {if a >= 0 {return a}return -a}func lowerBound(a []int, x int) int {idx := sort.Search(len(a), func(i int) bool {return a[i] >= x})return idx}func upperBound(a []int, x int) int {idx := sort.Search(len(a), func(i int) bool {return a[i] > x})return idx}//---------------------------------------------// priority queue//---------------------------------------------type pqi struct{ a int }type priorityQueue []pqifunc (pq priorityQueue) Len() int { return len(pq) }func (pq priorityQueue) Swap(i, j int) { pq[i], pq[j] = pq[j], pq[i] }func (pq priorityQueue) Less(i, j int) bool { return pq[i].a < pq[j].a }func (pq *priorityQueue) Push(x interface{}) { *pq = append(*pq, x.(pqi)) }func (pq *priorityQueue) Pop() interface{} {x := (*pq)[len(*pq)-1]*pq = (*pq)[0 : len(*pq)-1]return x}func main() {defer wr.Flush()sc.Split(bufio.ScanWords)sc.Buffer([]byte{}, 1000000)// this template is new version.// use getI(), getS(), getInts(), getF()N := getI()a := getInts(N)b := make([]int, N)pq := priorityQueue{}for i := 0; i < N; i++ {x := a[i] - 1if b[x] != 0 {heap.Push(&pq, pqi{a[i] - 1})}b[x]++}ans := 0for i := 0; i < len(b); i++ {if b[i] == 0 {x := heap.Pop(&pq).(pqi)ans += abs(i - x.a)}}out(ans)}