[SWEA D3] 1206 View (python 파이썬)

SW Expert Academy

---

처음 풀어본 SW Expert Academy. 아직 한 문제만 풀어봐서 잘 모르겠다. 몇 문제 풀어보면 느낌을 알 거 같다.

문제 페이지에 input.txt를 보고 파일 입출력을 해야 하나 했는데, 그냥 평소 문제 풀듯 풀면 된다.

 

근데 궁금한 게 input이 전부 주어지면, 하드코딩으로 제출해도 정답인가? 모르겠다.

 

그리고 이 문제만 그런지 모르겠는데, 출력이 특이했다. #1 1231, #2 5343, #3 1233... 이런 식으로

제출 환경도 처음 봐서 살짝 뇌가 정지했는데, 그냥 원래대로 하면 된다.

---

아이디어

코드 길이를 보면 알겠지만 매우 간단하다. 오히려 너무 쉬워서 의심을 많이 할 수도 있다.

아무리 생각해 봐도 tc 10개가 30초를 넘을 거 같진 않았다.

 

그냥 각 건물을 기준으로 +-2 범위에 있는 건물들 중 max값을 계산해서 빼주면 된다.

음수는 0으로 바꿔준다.

---

전체 코드

for t in range(1, 11):
    N = int(input())

    buildings = [int(x) for x in input().split()]

    Sum = 0
    for i in range(2, N - 2):
        ans = buildings[i] - max(buildings[i - 2:i] + buildings[i + 1: i + 3])
        if(ans < 0):
            ans = 0
        Sum += ans

    print('#', end = '')
    print(t, Sum)