[백준 21679] Клавиатура (C#, C++) - soo:bak

문제 링크

21679번 - Клавиатура

설명

각 키의 허용 누름 횟수와 실제로 눌린 키 목록이 주어질 때, 각 키가 고장났는지 여부를 출력하는 문제입니다.

접근법

각 키의 허용 횟수를 배열에 저장합니다.

눌린 키 목록을 순회하면서 해당 키의 허용 횟수를 1씩 감소시킵니다. 허용 횟수가 음수가 되면 허용 횟수를 초과한 것이므로 고장입니다.

모든 입력을 처리한 뒤, 각 키의 남은 허용 횟수가 음수이면 yes를, 아니면 no를 출력합니다.

---

Code

C#

using System;

class Program {
  static void Main() {
    var n = int.Parse(Console.ReadLine()!);
    var capStr = Console.ReadLine()!.Split();
    var cap = new int[n];
    for (var i = 0; i < n; i++)
      cap[i] = int.Parse(capStr[i]);

    var k = int.Parse(Console.ReadLine()!);
    var presses = Console.ReadLine()!.Split();
    for (var i = 0; i < k; i++) {
      var idx = int.Parse(presses[i]) - 1;
      cap[idx]--;
    }

    for (var i = 0; i < n; i++)
      Console.WriteLine(cap[i] < 0 ? "yes" : "no");
  }
}

C++

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

typedef vector<int> vi;

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

  int n; cin >> n;
  vi cap(n);
  for (int i = 0; i < n; i++)
    cin >> cap[i];

  int k; cin >> k;
  for (int i = 0; i < k; i++) {
    int idx; cin >> idx;
    cap[idx - 1]--;
  }

  for (int i = 0; i < n; i++) {
    if (cap[i] < 0) cout << "yes\n";
    else cout << "no\n";
  }

  return 0;
}