[Algorithm] 프로그래머스 - 문자열안에 문자열 (Kotlin)

문자열 str1, str2가 매개변수로 주어진다. str1 안에 str2가 있다면 1, 없다면 2를 리턴하는 solution()을 완성하라

 

 

아래는 주먹구구식으로 푼 코드다.

 

class Solution {
    fun solution(str1: String, str2: String): Int {
        var answer: Int = 0
        answer += if (str1.contains(str2)) {
            1
        } else {
            2
        }

        return answer
    }
}

 

이것을 간단하게 줄이면 아래와 같다.

 

class Solution {
    fun solution(str1: String, str2: String) = if (str1.contains(str2)) 1 else 2
}

 

또는 아래도 가능하다.

 

class Solution {
    fun solution(str1: String, str2: String) = if( str2 in str1 ) 1 else 2
}