[알고리즘] 38. Jumping on the Clouds: Revisited

Aerith is playing a cloud hopping game. In this game, there are sequentially numbered clouds that can be thunderheads or cumulus clouds. Her character must jump from cloud to cloud until it reaches the start again.

To play, Aerith is given an array of clouds,  and an energy level . She starts from  and uses  unit of energy to make a jump of size  to cloud . If Aerith lands on a thundercloud, , her energy () decreases by  additional units. The game ends when Aerith lands back on cloud .

Given the values of , , and the configuration of the clouds as an array , can you determine the final value of  after the game ends?

For example, give  and , the indices of her path are . Her energy level reduces by  for each jump to . She landed on one thunderhead at an additional cost of  energy units. Her final energy level is .

Note: Recall that  refers to the modulo operation. In this case, it serves to make the route circular. If Aerith is at  and jumps , she will arrive at .

Function Description

Complete the jumpingOnClouds function in the editor below. It should return an integer representing the energy level remaining after the game.

jumpingOnClouds has the following parameter(s):

• c: an array of integers representing cloud types
• k: an integer representing the length of one jump

Input Format

The first line contains two space-separated integers,  and , the number of clouds and the jump distance.
The second line contains  space-separated integers  where . Each cloud is described as follows:

• If , then cloud  is a cumulus cloud.
• If , then cloud  is a thunderhead.

Constraints

Output Format

Print the final value of  on a new line.

Sample Input

8 2
0 0 1 0 0 1 1 0

Sample Output

92

Explanation

In the diagram below, red clouds are thunderheads and purple clouds are cumulus clouds:

Observe that our thunderheads are the clouds numbered , , and . Aerith makes the following sequence of moves:

1. Move: , Energy: .
2. Move: , Energy: .
3. Move: , Energy: .
4. Move: , Energy: .

#!/bin/python3

import math

import os

import random

import re

import sys

# Complete the jumpingOnClouds function below.

def jumpingOnClouds(c, k):

energy = 100

new = list()

for i in range(n):

new.append((i+k)%n)

if (i+k)%n == 0:

break

for i in new[::k]:

if c[i] == 1:

energy -= 3

else:

energy -= 1

return energy

if __name__ == '__main__':

fptr = open(os.environ['OUTPUT_PATH'], 'w')

nk = input().split()

n = int(nk[0])

k = int(nk[1])

c = list(map(int, input().rstrip().split()))

result = jumpingOnClouds(c, k)

fptr.write(str(result) + '\n')

fptr.close()

점프했을 때, 도착한 값이 1일 경우 -3, 0일 경우 -1을 하여 100에서 부터 빼주어 진행! (i+k)%n으로 시작점을 잡고 시작한다. idx=0이 될 때까지의 리스트를 만들고 이를 인덱스로 사용했다.