1205 - 등수 구하기

https://www.acmicpc.net/problem/1205

1205번: 등수 구하기

생각해 보기

• N이 P보다 작은 경우
• 모든 배열과 비교했지만 답이 나오지 않은 경우
  • 마지막 배열 값과 동일한가?
  • 작으면 P 범위를 벗어나는지

예제

10 1 11
1 1 1 1 1 1 1 1 1 1
답: 1

3 70 10
100 90 80
답: 4

코드

#include<iostream>
#include<vector>

using namespace std;

int N,T,P;
vector<int> ranks;

void input(){
    cin>>N>>T>>P;
    ranks.resize(N+1);
    for(int i=0;i<N;++i){
        cin>>ranks[i];
    }
}

void solution(){
    int cnt=1;
    for(int i=0;i<N&&i<P;++i){
        if(ranks[i]<T){
            if(i>0&&ranks[i-1]>T)
                cnt=i+1;
            cout<<cnt<<'\n';
            return;
        }
        if(i>0&&ranks[i-1]>ranks[i])
            cnt=i+1;
    }
    
    if(N<P){
        if(ranks[N-1]==T)
            cout<<cnt<<'\n';
        else
            cout<<N+1<<'\n';
    } 
    else
        cout<<-1<<'\n';
}

int main(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);

    input();
    solution();

    return 0;
}