[백준 31496] 자작나무가 없소~ (C#, C++) - soo:bak

문제 링크

31496번 - 자작나무가 없소~

설명

아이템 이름을 구분자 _로 나눈 단어 중에 주어진 단어 S가 있으면 해당 아이템의 수량을 합산하는 문제입니다.

접근법

각 아이템 이름을 왼쪽부터 확인하면서 _로 구분된 단어를 비교합니다.
단어가 S와 일치하면 그 아이템의 수량을 더하고, 모든 아이템을 처리한 뒤 합을 출력합니다.

---

Code

C#

using System;
using System.Text;

class Program {
  static void Main() {
    var parts = Console.In.ReadToEnd().Split();
    var idx = 0;
    var n = int.Parse(parts[idx++]);
    var target = parts[idx++];

    var sum = 0;
    for (var i = 0; i < n; i++) {
      var name = parts[idx++];
      var cnt = int.Parse(parts[idx++]);

      var ok = false;
      var start = 0;
      for (var p = 0; p <= name.Length; p++) {
        if (p == name.Length || name[p] == '_') {
          if (p - start == target.Length && name.Substring(start, p - start) == target) {
            ok = true;
            break;
          }
          start = p + 1;
        }
      }

      if (ok) sum += cnt;
    }

    Console.Write(sum);
  }
}

C++

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

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

  int n; string target;
  cin >> n >> target;

  int sum = 0;
  for (int i = 0; i < n; i++) {
    string name; int cnt;
    cin >> name >> cnt;

    bool ok = false;
    int start = 0;
    for (int p = 0; p <= (int)name.size(); p++) {
      if (p == (int)name.size() || name[p] == '_') {
        if (p - start == (int)target.size() && name.compare(start, p - start, target) == 0) {
          ok = true;
          break;
        }
        start = p + 1;
      }
    }

    if (ok) sum += cnt;
  }

  cout << sum << "\n";

  return 0;
}