파이썬 알고리즘 099 | [백준 2775번] 부녀회장이 될테야

99. 부녀회장이 될테야

문제
평소 반상회에 참석하는 것을 좋아하는 주희는 이번 기회에 부녀회장이 되고 싶어 각 층의 사람들을 불러 모아 반상회를 주최하려고 한다.

이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다.

아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라. 단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.

입력
첫 번째 줄에 Test case의 수 T가 주어진다. 그리고 각각의 케이스마다 입력으로 첫 번째 줄에 정수 k, 두 번째 줄에 정수 n이 주어진다

출력
각각의 Test case에 대해서 해당 집에 거주민 수를 출력하라.

제한
1 ≤ k, n ≤ 14
예제 입력 1
2
1
3
2
3
예제 출력 1
6
10

<내 풀이>


t=int(input())

for _ in range(t) : 
    v=1
    k=int(input())
    n=int(input())
    arr = [[0 for col in range(n+1)] for row in range(k+1)]
    for j in range(1,n+1) :
        arr[0][j]=v
        v+=1

    for i in range(1,k+1) :
        for j in range(1,n+1):
            if j==1 : arr[i][j]=1
            else : arr[i][j] = arr[i][j-1] + arr[i-1][j]

    print(arr[k][n])

=> 가장 단순한 방법으로 접근했다
=> 우선 0층의 사람들을 배열에 넣어놓은 다음
=> 차례차례로 1층 사람들의 수를 0층 사람들 토대로..
=> 2층 사람들의 수를 1층 사람들 토대로..
차곡차곡 채워가다보면 k층n호까지의 사람들이 구해짐

<틀린 풀이>


t=int(input())
v=1
for _ in range(t) : 
    k=int(input())
    n=int(input())
    arr = [[0 for col in range(n+1)] for row in range(k+1)]
    for j in range(1,n+1) :
        arr[0][j]=v
        v+=1

    for i in range(1,k+1) :
        for j in range(1,n+1):
            if j==1 : arr[i][j]=1
            else : arr[i][j] = arr[i][j-1] + arr[i-1][j]

    for i in range(1,k+1) : 
        print("\n")
        for j in range(1,n+1) :
            print(arr[i][j], end=' ')

    print(arr[k][n])

=> 어디서 오류가 나는건지 한참을 고민했는데 코드 논리자체의 오류가 아니었다.. ~~
~~바로 내가 지금까지 야매로 써오고 있던 테스트 케이스 입력을 받는 것에 관한 문제였던 거시닷!!!!!!!!! 아래의 반성점에 작성하겠다

=> 아니다..

<반성 점>

(1) 처음에 내가 잘못 파악한 원인

~~

=> 위는 정답 코드고 아래가 나의 코드이다.
이렇게 하나 입력 - 하나 출력을 하면 결과가 the same 한 것을 확인할 수 있음
그러나 위와 동일한 입력을 testcase를 두개 준다고 가정하고 넣어보겠다.

이렇게 된다

노란색 표시를 보면 다르다~~

(2) 궁극적인 원인은 내 기존 오답 코드에서는 v=1을 ...testcode를 돌리기 전에 선언해놔서, 이게 test code가 다시 시작될 때는 v=1로 갱신되어 있어야 하는데, 갱신이 되지 않고 있던 것..!!!!!!!!!!!!!!!!!!

반성점 : 변수 선언 위치 절대 중요해 절대 지켜!!!!!!!!!!

<배운 점>

    arr = [[0 for col in range(n+1)] for row in range(k+1)]
  • 2차원 배열 선언 방법(1)
>>> a = [[0 for j in range(2)] for i in range(3)]
>>> a
[[0, 0], [0, 0], [0, 0]]
  • 2차원 배열 선언 방법(2)
>>> a = [[0] * 2 for i in range(3)]
>>> a
[[0, 0], [0, 0], [0, 0]]

=> col 자리에 행, row 자리에 열 갯수

  • python에서 가로로 print 하는 방법 print("arr[k][n]", end=' ')

  • (+) 복습 range(시작범위, 끝범위-1)

좋은 웹페이지 즐겨찾기