본문 바로가기

백준

8979 올림픽

이걸 처음에 틀려서 어느 부분이 문제인지 몰라서 엄청 해맸는데

 

알고보니 벡터 사이즈를 v(n + 5)로 하고 sort(v.begin(), v.end()); 라고 해서 문제였다.

 

이 부분이 문제일꺼라 상상도 못함ㅋㅋㅋㅋ 직접 만든 정렬 함수가 문제인줄 알았다고..

 

그래서 이거 사이즈 n으로 맞추니깐 통과했다

 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <functional>
#include <string>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <cmath>
#include <cstring>
#include <bitset>

#define xx first
#define yy second
#define all(x) (x).begin(), (x).end()

using namespace std;
using i64 = long long int;
using ii = pair<int, int>;
using ii64 = pair<i64, i64>;
using iii = tuple<int, int, int>;

using namespace std;

struct st {
    int n;
    int g;
    int s;
    int b;
};

bool sort_by_struct(struct st a, struct st b)
{
    if (a.g == b.g && a.s == b.s)
        return a.b > b.b;
    if (a.g == b.g)
        return a.s > b.s;
    return a.g > b.g;
}

int     main()
{
    int n, k;
    scanf("%d %d", &n, &k);

    vector<struct st> v(n);
    for (int i = 0; i < n; i++)
        cin >> v[i].n >> v[i].g >> v[i].s >> v[i].b;

    sort(v.begin(), v.end(), sort_by_struct);

    int count = 1;

    if (v[0].n == k)
    {
        printf("%d\n", count);
        return 0;
    }

    for (int i = 1; i < n; i++)
    {
        if (v[i].g != v[i - 1].g ||
            v[i].s != v[i - 1].s ||
            v[i].b != v[i - 1].b)
            count = i + 1;

        if (v[i].n == k)
        {
            printf("%d\n", count);
            return 0;
        }
    }
    
    return 0;
}

'백준' 카테고리의 다른 글

2138 전구와 스위치  (0) 2020.10.06
2877 4와 7  (0) 2020.10.06
11656 접미사 배열  (0) 2020.10.06
14624 전북대학교  (0) 2020.10.06
17300 패턴 [미완]  (0) 2020.10.04