[백준 29454] Выражение (C#, C++) - soo:bak

문제 링크

29454번 - Выражение

설명

n개의 정수 중 나머지 원소들의 합과 같은 원소의 인덱스를 찾는 문제입니다.

접근법

어떤 원소가 나머지 합과 같다면, 전체 합에서 그 원소를 뺀 값이 그 원소와 같아야 합니다. 전체 합을 먼저 구한 뒤 각 원소를 확인하면서 조건을 만족하는 첫 인덱스를 출력합니다.

---

Code

C#

using System;

class Program {
  static void Main() {
    var n = int.Parse(Console.ReadLine()!);
    var arr = Array.ConvertAll(Console.ReadLine()!.Split(), int.Parse);

    var total = 0;
    foreach (var v in arr) total += v;

    for (var i = 0; i < n; i++) {
      if (total - arr[i] == arr[i]) {
        Console.WriteLine(i + 1);
        break;
      }
    }
  }
}

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 a(n);
  int total = 0;
  for (int i = 0; i < n; i++) {
    cin >> a[i];
    total += a[i];
  }

  for (int i = 0; i < n; i++) {
    if (total - a[i] == a[i]) {
      cout << i + 1 << "\n";
      break;
    }
  }

  return 0;
}