[python] SWEA - 4301. 콩 많이 심기

🤔문제 해결

O	O	X	X	O	O
O	O	X	X	O	O
X	X	O	O	X	X
X	X	O	O	X	X
O	O	X	X	O	O
O	O	X	X	O	O

 

O	O	X	X	O
O	O	X	X	O
X	X	O	O	X

 

위에서부터 아래로 한줄씩 내려가면서 규칙을 찾아보면위의 그림처럼 심는게 가장 경우의 수가 많게 심을 수 있다. 

• 첫줄에서
  • 오른쪽으로 4칸마다 2개씩 심을 수 있다
  • 그리고 나머지칸이
    • 3칸 - 2개
    • 2칸 - 2개
    • 1칸 - 1개
    • 0칸 - 0개
• 두번째 줄도 위와 같다
• 세번째 줄은 앞의 두칸을 빼고 시작하여 위의 규칙을 적용
• 네번째도 세번째와 같다
• 다섯번째는 다시 첫번째처럼

결론은 1,2 | 3,4 | 5,6 | 7,8 | ... 짝지어서 하면된다.

💻소스 코드

def get_axis_per_count(num):
    count = num // 4 * 2
    nmg = num % 4
    if nmg in (3, 2):
        count += 2
    else:
        count += nmg
    return count


T = int(input())
for test_case in range(1, T + 1):
    N, M = map(int, input().strip().split())

    answer = 0
    flag = 0
    for i in range(M):
        if flag in (0, 1):
            answer += get_axis_per_count(N)
            flag += 1
        else:
            answer += get_axis_per_count(N - 2)
            flag += 1
        flag %= 4
    print(f'#{test_case} {answer}')

📕문제 확인

출처: SW Expert Academy