36-H0ngJu

[H0ngJu]

🔗 문제 링크

머리 톡톡

✔️ 소요된 시간

1시간

✨ 수도 코드

문제 자체는 굉장히 간단합니다.

원으로 앉은 학생들이 한명씩 일어나서 자신의 수가 앉아있는 사람의 배수 이면 머리를 톡톡치고, 총 머리를 때린 학생의 수를 반환하면 됩니다.

문제는 N(1≤N≤100,000)명 이고, 학생의 머리에 쓴 수는 1≤student≤1,000,000 라는 것입니다.

완탐으로 돌리면 시간초과가 발생합니다..

그러면 어떻게 접근할 것인가...

생각을 해봤는데요

1. n*m은 n의 배수임과 m의 배수임을 이용한다.
2. 때린 정보를 저장해두었다가 1.을 이용하여 더해준다.

이렇게 조합해서 여러 시도를 하다보니,

약수를 구해서 해당 약수를 적은 학생의 수만큼 때리도록 구현하였어요.

N = int(input())
students = [int(input()) for _ in range(N)]
hits = [0 for _ in range(1000001)] # 숫자의 개수
results = [0 for _ in range(N)]

for student in students: # 숫자 k를 적은 학생의 수를 기록
    hits[student] += 1

for idx in range(len(students)):
    for i in range(1, int(students[idx]**(1/2)) + 1):
        if students[idx] % i == 0:
            results[idx] += hits[i] # 학생 수 만큼 때리기
            if (i ** 2) != students[idx]: # 제곱이 아닌 경우
                results[idx] +=  hits[students[idx] // i]

print(hits)
for r in results:
    print(r-1, end=" ")

주의해야 할 부분!!

학생이 머리위에 쓴 수는 중복될 수 있습니다.
그리고 자신 빼고 때린다는 점을 주의해야해요

students = [2, 2, 3, 1]
이라고 되어 있으면 students[0]은 students[1]과 students[3]을 때릴 수 있으므로 2번 때릴 수 있습니다.

자신 빼고 때린다 -> 일단 다 때린 횟수를 구하고 자기 자신(1)을 뺀다

는 로직으로 구현하는 것이 더 쉬운 것 같아요.

약수 구하는 과정에서 students 안에 자신과 같은 사람이 있는지
더블체크 하는 순간 또 시간초과가 발생하기 때문입니다...

다 풀고 나니까 코드는 간단한데
시간초과를 탈출하느라 힘들었습니다...

📚 새롭게 알게된 내용

[tgyuuAn]

import kotlin.math.*;

fun main() {
    val N = readln().toInt()
    val answer = HashMap<Int, Int>()
    val counter = HashMap<Int, Int>()
    val numbers = mutableListOf<Int>()
    val forCheck = mutableSetOf<Int>()
    
    repeat(N) { idx ->
        val now = readln().toInt()
        counter[now] = (counter[now] ?: 0) + 1
        numbers.add(now)
        forCheck.add(now)
    }
    
    for(number in numbers) {
        if(answer[number] == null) {
            for(i in 1 until (sqrt(number.toFloat()).toInt()+1)) {
                if(number%i==0) {
                    answer[number] = (answer[number]?:0) + (counter[i] ?: 0)
                
                    if(i != number/i){
                        answer[number] = (answer[number]?:0) + (counter[number/i] ?: 0)
                    }
                }
            }
        }
        
        println((answer[number]?:1)-1)
    }
}

와 요즘 코틀린으로만 풀려고 하는데,

풀이는 떠오르는데 구현하는데 시간이 너무 오래걸리네요,.,,,

부윀,,,

이 문제 생각보다 시간 많이걸렸습니다...

리마인드 감사합니당 ㅎㅎㅎㅎㅎ

++

자신 빼고 때린다 -> 일단 다 때린 횟수를 구하고 자기 자신(1)을 뺀다

저는 써주신 것 처럼 이렇게 구했습니다!!