Kotlin/Kotlin 알고리즘

백준 26026번 Coffee Cup Combo Kotlin 구현해보기

kimc 2022. 11. 19. 00:00

```

백준 26026번 Coffee Cup Combo Kotlin 구현해보기

```

Kimc Kotlin Study

이번 글을 통해 배워갈 내용

  1. 백준 26026번 Coffee Cup Combo Kotlin 풀이

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

 

26026번: Coffee Cup Combo

Jonna is a university student who attends $n$ lectures every day. Since most lectures are way too simple for an algorithmic expert such as Jonna, she can only stay awake during a lecture if she is drinking coffee. During a single lecture she needs to drink

www.acmicpc.net

 

백준 26026번 Coffee Cup Combo는

 

 

난이도 브론즈 등급의 문제로서

 

N개의 강의실에서 각 강의실만큼 강의를 순서대로 듣고

강의실 앞에는 커피머신이 있어서

커피머신에서 한잔 마시고 2개의 커피를 들고 다닐 수 있다고 가정할 때

1은 커피머신이 있는 강의실

0은 커피머신이 없는 강의실이고

0과 1로 이루어진 강의실 이용 리스트가 주어질 때

강의마다 한번 커피를 마시면 깨어있다고 가정할 때

깨어있는 횟수를 구해주면 되는 문제입니다.


30분 정도 위에 링크를 방문하셔서 풀어보시고

안 풀리시는 경우에만 아래 해답을 봐주시면 감사하겠습니다.


강의 순서대로 돌면서

커피머신이 있으면

커피의 수 = 2

깨어있는 숫자 +1

 

커피머신이 없고 커피가 있으면

커피의수 -1

깨어있는 숫자 +1

해주고

깨어있는 숫자를 반환해주면 됩니다.

fun main(args: Array<String>) {
    // input
    val lectureCnt = readln().toInt()
    val lectureHallMap = readln()
    print(findMaxAwakeLectureCnt(lectureCnt, lectureHallMap))
}

fun findMaxAwakeLectureCnt(lectureCnt: Int, lectureHallMap: String): Int {

    var coffeeStock = 0
    var lectureAwakeCnt = 0

    for (i in 0 until lectureCnt) {
        if (lectureHallMap[i] == '1') {
            coffeeStock = 2
            lectureAwakeCnt++
        } else {
            if (coffeeStock > 0) {
                coffeeStock--
                lectureAwakeCnt++
            }
        }
    }

    return lectureAwakeCnt
}

// https://codemasterkimc.tistory.com/

 

테스트

import org.junit.jupiter.api.Assertions.assertEquals
import org.junit.jupiter.api.Test

internal class MainKtTest {

    @Test
    fun test() {
        assertEquals(8, findMaxAwakeLectureCnt(10, "1001000010"));
        assertEquals(4, findMaxAwakeLectureCnt(10, "1100000000"));
        assertEquals(0, findMaxAwakeLectureCnt(1, "0"));
    }
}

 

 

 

읽어주셔서 감사합니다

 

무엇인가 얻어가셨기를 바라며

 

오늘도 즐거운 코딩 하시길 바랍니다 ~ :)

 


 

728x90