Advertisement
paranid5

kotlin_gay

Jan 24th, 2021
220
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Kotlin 1.09 KB | None | 0 0
  1. fun main() {
  2.     if ((readLine() ?: return).toInt() < 5) {
  3.         print("No")
  4.     } else {
  5.         val arr = (readLine() ?: return).split(' ').map { x -> x.toLong() }.sorted()
  6.         val sum = arr.sum()
  7.  
  8.         if (sum % 5 != 0L) {
  9.             print("No")
  10.         } else {
  11.             val subSum = sum / 5
  12.             var sub = 0L
  13.  
  14.             for (i in arr) {
  15.                 if (subSum > 0) {
  16.                     when {
  17.                         sub + i < subSum -> sub += i
  18.                         sub + i == subSum -> sub = 0
  19.                         else -> {
  20.                             print("No")
  21.                             return
  22.                         }
  23.                     }
  24.                 } else {
  25.                     when {
  26.                         sub + i > subSum -> sub += i
  27.                         sub + i == subSum -> sub = 0
  28.                         else -> {
  29.                             print("No")
  30.                             return
  31.                         }
  32.                     }
  33.                 }
  34.             }
  35.  
  36.             print("Yes")
  37.         }
  38.     }
  39. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement