글 읽기 - Golang 입출력 관련하여 질문드립니다!

snkii35 2년 전 0

시간초과가 반복하여 뜨는데, 원인이 입출력 시스템에 있는 것으로 추측됩니다.

관련하여 조언 부탁드립니다.

감사합니다!

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
	"strings"
)

func main() {
	var heap = []int{}
	rd := bufio.NewReader(os.Stdin)
	wt := bufio.NewWriter(os.Stdout)
	defer wt.Flush()
	n, _ := rd.ReadString('\n')
	n = strings.TrimSpace(n)
	n_int, _ := strconv.Atoi(n)
	for i := 0; i < n_int; i++ {
		n, _ = rd.ReadString('\n')
		n = strings.TrimSpace(n)
		val, _ := strconv.Atoi(n)
		if val == 0 {
			// print max value(root)
			// fmt.Fprint(wt, "Output: ")
			if len(heap) == 0 {
				fmt.Fprint(wt, 0, "\n")
			} else {
				fmt.Fprint(wt, heap[0], "\n")
				heap = heap[1:]
			}
		} else {
			heap = append(heap, val)
			// max-heapify
			for i := len(heap)/2 - 1; i >= 0; i-- {
				MaxHeapify(heap, i)
			}
		}
	}
}

func MaxHeapify(heap []int, i int) {
	left := 2 * (i+1) -1
	right := 2 * (i+1)
	largest := i
	if left <= len(heap)-1 && heap[left] > heap[i] {
		largest = left
	}
	if right <= len(heap)-1 && heap[right] > heap[i] {
		largest = right
	}
	if largest != i {
		heap[i], heap[largest] = heap[largest], heap[i]
		MaxHeapify(heap, largest)
	}
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

Golang 입출력 관련하여 질문드립니다!

snkii35 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말