g0101_0200.s0135_candy.Solution.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of leetcode-in-kotlin Show documentation
Show all versions of leetcode-in-kotlin Show documentation
Kotlin-based LeetCode algorithm problem solutions, regularly updated
package g0101_0200.s0135_candy
// #Hard #Array #Greedy #2022_10_09_Time_466_ms_(58.33%)_Space_51.2_MB_(83.33%)
import java.util.Arrays
class Solution {
fun candy(ratings: IntArray): Int {
val candies = IntArray(ratings.size)
Arrays.fill(candies, 1)
for (i in 0 until ratings.size - 1) {
if (ratings[i + 1] > ratings[i]) {
candies[i + 1] = candies[i] + 1
}
}
for (i in ratings.size - 1 downTo 1) {
if (ratings[i - 1] > ratings[i] && candies[i - 1] < candies[i] + 1) {
candies[i - 1] = candies[i] + 1
}
}
var total = 0
for (candy in candies) {
total += candy
}
return total
}
}