Advertisement
saurav_kalsoor

Help Mocha - KOTLIN

Sep 22nd, 2021
90
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Kotlin 1.11 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 likes = Array(n) { IntArray(2) }
  10.     for (i in 0 until n) {
  11.         likes[i][0] = sc.nextInt()
  12.         likes[i][1] = sc.nextInt()
  13.     }
  14.  
  15.     val resultArray = helpMocha(likes, n)
  16.     for (a in resultArray) print("$a ")
  17.     println()
  18. }
  19.  
  20.  
  21. fun helpMocha(likes: Array<IntArray>, n: Int): IntArray {
  22.     val mp = HashMap<Int, HashSet<Int>>()
  23.     val resultSet = HashSet<Int>()
  24.     for (like in likes) {
  25.         if (!mp.containsKey(like[0])) {
  26.             mp[like[0]] = HashSet()
  27.         }
  28.         val temp = mp[like[0]]!!
  29.         temp.add(like[1])
  30.         mp[like[0]] = temp
  31.         if (mp.containsKey(like[1]) && mp[like[1]]!!.contains(like[0])) {
  32.             resultSet.add(like[0])
  33.             resultSet.add(like[1])
  34.         }
  35.     }
  36.     if (resultSet.isEmpty()) return intArrayOf(-1)
  37.     val resultArray = IntArray(resultSet.size)
  38.     var i = 0
  39.     for (a in resultSet) {
  40.         resultArray[i] = a
  41.         i++
  42.     }
  43.     Arrays.sort(resultArray)
  44.     return resultArray
  45. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement