[백준] 파이썬 python - 2841번 : 외계인의 기타 연주

https://www.acmicpc.net/problem/2841

2841번: 외계인의 기타 연주

2841번: 외계인의 기타 연주

레벨 실버 1

 

 

접근 방식
1. 총 6개의 줄이므로, 각각에 해당하는 스택을 미리 만든다.
2. line에 해당하는 스택이 존재하며, top에 있는 프렛이 현재 누를 프렛보다 높다면 해당 프렛 삭제 (손 떼어내기)
3. 이미 해당 프렛을 누르고 있다면, continue
4. 현 프렛을 stack에 추가 (누르기)

 

 

내 코드

n, P = map(int, input().split())

stk = [[] for _ in range(7)] # 총 6개의 줄
answer=0

for i in range(n) :
    line, p = map(int ,input().split())
    
    if len(stk[line]) == 0 : #존재하지 않으면
        stk[line].append(p)
        answer+=1
        
    else: #존재하면
        if stk[line][-1] < p : # 현재 음이 더 높으면
            stk[line].append(p) # 누르기
            answer+=1
        
        elif stk[line][-1] > p : # 떼어낼 필요가 있는 경우
            for _ in range(len(stk[line])) :

                if stk[line][-1] < p : #반복하다가, 떼어낼 필요 없으면 break
                    break
                
                if stk[line][-1] > p : # 손 떼어내기               
                    answer+=1
                    stk[line].pop()
                
                elif stk[line][-1] == p: # 현 프렛과 동일하면 break
                    break
            
            if len(stk[line]) == 0: # 누르기
                stk[line].append(p)
                answer+=1
            
            elif stk[line][-1] != p: # 누르기
                stk[line].append(p)
                answer+=1
            
            elif stk[line][-1] == p :
                continue
                
        elif stk[line][-1] == p :
            continue
        
        
print(answer)

주구장창 써놓긴 했지만 별 건 아니고 길게 썼다.. 그냥 문제를 풀다보니까 어 이 경우 안 되네 저 경우 안되네 하다가 코드가 길어짐 == 최적화가 되지 않았음..^^ 

물론 아래의 답과 성능상은 크게 다르지 않겠지만 직관성이 떨어진다.

 

 

최종

N, P = map(int, input().split())

stk = [[] for _ in range(7)] # 총 6개의 줄
ans = 0

for _ in range(N) :
    line, p = map(int, input().split())
    
    #해당 line이 존재하고, 현 프렛보다 높다면
    while stk[line] and stk[line][-1] > p :
        stk[line].pop()
        ans+=1
        
    # 이미 해당 프렛을 누르고 있다면 continue
    if stk[line] and stk[line][-1] == p:
        continue
    
    stk[line].append(p)
    ans+=1
    
print(ans)

 

내 코드를 다 작성하고, 맞은 걸 본 후에 답을 찾아봤다.

내 말도 안되게 쓸데없이 긴 코드를 깔끔하게 추려보자면 위와 같은 코드로 결론날 수 있을 것이다.

예제에 답을 끼워넣다보니 저렇게 길어지는 것 같다. 나중에는 코드를 최적화할 수 있는 방향에 대해서 고민해봐야 할 것 같다.

 

내가 작성한 코드는 백준에 돌려봤을 때 448ms, 아래의 코드는 396ms로 나타났다.