글 읽기 - 반례나 지적 부탁드립니다.

yhj1937 2년 전 0

'틀렸습니다'가 나온 풀이입니다.

#include <bits/stdc++.h>
using namespace std;

int p, l, g, x;
pair<int, int> p2lg[100005];  // 문제 번호에서 난이도와 알고리즘 분류를 얻는 변수
set<int> l2p[105];  // 난이도마다 문제 번호를 저장하는 변수
multiset<int> g2l[105];  // 알고리즘 분류마다 난이도를 저장하는 변수

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

  int n;
  cin >> n;
  while (n--) {
    cin >> p >> l >> g;
    p2lg[p] = {l, g};
    l2p[l].insert(p);
    g2l[g].insert(l);
  }

  int m;
  cin >> m;
  while (m--) {
    string com;
    cin >> com;
    if (com == "recommend") {
      cin >> g >> x;
      // g2l에서 가장 어려운/쉬운 난이도를 구하고
      // 그 난이도로 l2p에서 가장 큰/작은 문제 번호를 구함
      if (x == 1) {
        l = *prev(g2l[g].end());
        cout << *prev(l2p[l].end()) << '\n';
      } else {
        l = *g2l[g].begin();
        cout << *l2p[l].begin() << '\n';
      }
    } else if (com == "recommend2") {
      cin >> x;
      // l2p를 100부터 아래 혹은 1부터 위로 탐색
      if (x == 1) {
        for (l = 100; l >= 1; --l)
          if (!l2p[l].empty()) break;
        cout << *prev(l2p[l].end()) << '\n';
      } else {
        for (l = 1; l <= 100; ++l)
          if (!l2p[l].empty()) break;
        cout << *l2p[l].begin() << '\n';
      }
    } else if (com == "recommend3") {
      cin >> x >> l;
      // l2p를 l부터 위 혹은 l-1부터 아래로 탐색
      if (x == 1) {
        for (; l <= 100; ++l)
          if (!l2p[l].empty()) break;
        if (l <= 100) cout << *l2p[l].begin() << '\n';
        else cout << -1 << '\n';
      } else {
        for (l -= 1; l >= 1; --l)
          if (!l2p[l].empty()) break;
        if (l >= 1) cout << *prev(l2p[l].end()) << '\n';
        else cout << -1 << '\n';
      }
    } else if (com == "add") {
      cin >> p >> l >> g;
      p2lg[p] = {l, g};
      l2p[l].insert(p);
      g2l[g].insert(l);
    } else {
      cin >> p;
      tie(l, g) = p2lg[p];
      g2l[g].erase(g2l[g].find(l)); // 하나만 삭제
      l2p[l].erase(p);
    }
  }
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

반례나 지적 부탁드립니다.

yhj1937 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말