SWEA - 전기버스 [Python]

😅 문제 

 

- https://swexpertacademy.com/main/

SW Expert Academy

 

🤔 문제 상황 

- 전기버스의 1회 충전시 이동 최대거리를 활용해 최소 횟수로 충전하여 목적지에 도달하는 방법의 수를 계산한다.

 

🧐 해결 전략 

- idx를 갱신하며 문제에 주어진 조건을 시뮬레이션하여 코드를 작성한다.

 

🎰 코드 

T = int(input())
# 결과를 담을 리스트
result = []
for _ in range(T):
    # K = 한번 충전으로 움직일 수 있는 최대 거리, N = 정류장 수, M = 충전기가 설치된 정류장 수
    K,N,M = map(int, input().split())
    # 충전기가 설치된 정류장
    charger_stop = list(map(int, input().split()))
    # 충전 횟수를 담을 변수
    time = 0
    # 반복을 통해 충전 횟수 찾기
    idx = 0
    while True:
        # 탈출 조건 형성(idx가 N보다 크거나 같으면 종료)
        if idx + K >= N:
            break
        # idx부터 idx + K 사이에 충전기가 있는 정류장 찾기
        is_Check = False
        for charger in range(idx+K, idx, -1):
            # charger가 주어진 정류장 안에 있을 경우
            if charger in charger_stop:
                idx = charger
                time += 1
                is_Check = True
                break
        # charger가 정류장에 없을 경우 -1을 결과로 하고 반복을 종료한다.
        
        if not is_Check:
            result.append(0)
            break
    if is_Check:
        result.append(time)
 
for i in range(T):
    print(f"#{i+1} {result[i]}")
 

- 기존 풀이

 

 
def findidx(l):
    idx = -1
    for i in range(len(l)):
        if l[i] == 1:
            idx = i
    return idx 
 
T = int(input())
result = []
def findcnt(ll):
    cnt = 0 # 방문 횟수
    i = 0
    while True:
        if i+k+1 <= n+1: 
            temp = findidx(ll[i:i+k+1]) + i # 갈수있는 거리 안에서 가장 먼 곳에 있는 충전소 위치
        else :
            temp = findidx(ll[i:]) + i
        if temp - i == -1:
            result.append(0)
            return
        else :
            charge[i] = 0
            i = temp
            cnt += 1
        if i + k >= n:
            break
    result.append(cnt)
 
for _ in range(T):
    k,n,m = map(int,input().split()) # k 최소 이동거리, n 버스 정류장 수, m 충전소 수
    mlist = list(map(int,input().split())) # 충전소 위치
    charge = [0 if i not in mlist else 1 for i in range(n+1)] # 버스정류장 
    findcnt(charge)
for i in range(T):
    print("#{} {}".format(i+1, result[i]))

🧙‍♂️ 해설

- 기본적인 메커니즘은 비슷하지만 불필요하게 많은 함수를 생성하고 list slicing을 사용하였다. 큰 속도 차이는 아니지만 오히려 가독이 힘든 코드였다. 

 

📈 새로 학습한 것 & 실수 

- 문제에 제시된 예시가 최선의 상황이 아니였다. 그래서 혼돈이 있었다. 

 

출처 - https://swexpertacademy.com/main/learn/course/lectureProblemViewer.do