파이썬 알고리즘 수업 - 알고리즘의 수행 시간 4(백준 BOJ 24265)


문제

오늘도 서준이는 알고리즘의 수행시간 수업 조교를 하고 있다. 아빠가 수업한 내용을 학생들이 잘 이해했는지 문제를 통해서 확인해보자.

입력의 크기 n이 주어지면 MenOfPassion 알고리즘 수행 시간을 예제 출력과 같은 방식으로 출력해보자.

MenOfPassion 알고리즘은 다음과 같다.

MenOfPassion(A[], n) {
    sum <- 0;
    for i <- 1 to n - 1
        for j <- i + 1 to n
            sum <- sum + A[i] × A[j]; # 코드1
    return sum;
}


입력

첫째 줄에 입력의 크기 n(1 ≤ n ≤ 500,000)이 주어진다.


출력

첫째 줄에 코드1 의 수행 횟수를 출력한다.

둘째 줄에 코드1의 수행 횟수를 다항식으로 나타내었을 때, 최고차항의 차수를 출력한다. 단, 다항식으로 나타낼 수 없거나 최고차항의 차수가 3보다 크면 4를 출력한다.


예제 입력 1

7

예제 출력 1

21
2

코드1 이 21회 수행되고 알고리즘의 수행 시간이 n2에 비례한다.


📝풀어보기

# MenOfPassion(A[], n) {
#     sum <- 0;
#     for i <- 1 to n - 1
#         for j <- i + 1 to n
#             sum <- sum + A[i] × A[j]; # 코드1
#     return sum;
# }

# 동작 확인
def MenOfPassion(A, n):
    sum = 0
    for i in range(1, n): 
        A.append(1)
        for j in range(i+1, n+1):
            A.append(1)

    for i in range(1, n):
        for j in range(i+1, n+1):
            sum += A[i] * A[j]
            
    return sum

A = []
n = int(input())
print(MenOfPassion(A, n))
# i는 [1, n-1], j는 [i+1, n] 이다. 
# i가 n-1번 도는 동안 j는 순서대로 n-1, n-2, n-3, ... , 1번 돌게 된다. 
# 따라서 답은 n-1 + n-2 + ... + 1 이 된다. 
# 등차가 1인 등차수열의 합을 구하면 되므로 첫째 줄에는 n*(n-1)/2 를 출력해주면 된다.

# 제출 코드
print(int(n*(n-1)/2))
print(2)

관심있을 포스팅