Advertisement
saurav_kalsoor

Find Power - KOTLIN

Sep 26th, 2021
151
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Kotlin 0.59 KB | None | 0 0
  1. package com.company
  2.  
  3. import java.util.*
  4.  
  5.  
  6. fun main() {
  7.     val sc = Scanner(System.`in`)
  8.     val n = sc.nextInt()
  9.     val power = IntArray(n)
  10.     for (i in 0 until n) {
  11.         power[i] = sc.nextInt()
  12.     }
  13.     val result = findPower(power, n)
  14.     println(result)
  15. }
  16.  
  17.  
  18. fun findPower(power: IntArray, n: Int): Int {
  19.     val pq = PriorityQueue(Collections.reverseOrder<Int>())
  20.     for (a in power)
  21.         pq.add(a)
  22.     while (pq.size > 1) {
  23.         val a = pq.poll()
  24.         val b = pq.poll()
  25.         if (a == b) pq.add(a) else pq.add((a + b) / 4)
  26.     }
  27.     return pq.poll()
  28. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement