[JAVA] 알고리즘 문제풀이 입문 5-8

강의: https://inf.run/w779

자바(Java) 알고리즘 문제풀이 입문: 코딩테스트 대비 - 인프런 | 강의

 

5-8번

import java.util.LinkedList;

import java.util.Queue;

import java.util.Scanner;

class Person{

  int id;

  int pri;

  public Person(int id, int pri){

    this.id = id;

    this.pri = pri;

  }

}

public class Main {

  public int solution(int n, int m, int[] arr){

    Queue<Person> que = new LinkedList<>();

    int answer = 1;

    for(int i = 0; i < n; i++){

      que.add(new Person(i, arr[i]));

    }

    while(!que.isEmpty()){

      Person p = que.poll();

      for(Person x : que){

        if(x.pri > p.pri){

          que.offer(p);

          p = null;

          break;

        }

      }

      if(p != null){

        if(p.id == m) return answer;

        else answer++;

      }

    }

    return answer;

  }

  public static void main(String[] args){

    Main t = new Main();

    Scanner in = new Scanner(System.in);

    int n = in.nextInt();

    int m = in.nextInt();

    int[] arr = new int[n];

    for(int i = 0; i < n; i++){

      arr[i] = in.nextInt();

    }

    in.close();

      System.out.print(t.solution(n, m, arr));

  }

}