[HackerRank] Jumping on the Clouds

[문제 링크]

[입력]

int c[n]: 경로에 따른 구름의 종류
int k: 점프 1번 진행할 때 가는 거리

[출력]

int: 다시 시작점으로 돌아왔을 때 남아있는 에너지의 양

[코드]

def jumpingOnClouds(c, k):
    energy = 100
    pos =0
    while 1:
        if((pos+k)%len(c)==0):
            if(c[0]==0):
                energy-=1
            else:
                energy-=3
            break
        else:
            if (c[(pos+k)%len(c)]==0):
                energy -=1
            else:
                energy -=3
            pos = (pos+k)%len(c)
    return energy