[백준 19774] ABCD-код (C#, C++) - soo:bak

문제 링크

19774번 - ABCD-код

설명

4자리 코드 ABCD가 주어집니다. 앞 두 자리 AB를 수로 보고 제곱한 값과 뒤 두 자리 CD를 수로 보고 제곱한 값의 합이 7로 나눈 나머지가 1이면 유효한 코드입니다.

각 후보에 대해 유효하면 YES, 아니면 NO를 출력하는 문제입니다.

접근법

먼저, 후보 문자열을 읽어 앞 두 글자와 뒤 두 글자를 분리하여 정수로 변환합니다.

다음으로, 두 수를 각각 제곱한 후 더한 값을 7로 나눈 나머지를 구합니다.

이후, 나머지가 1이면 YES, 아니면 NO를 출력합니다.

Code

C#

using System;

namespace Solution {
  class Program {
    static void Main(string[] args) {
      var t = int.Parse(Console.ReadLine()!);
      for (var i = 0; i < t; i++) {
        var s = Console.ReadLine()!;
        var front = int.Parse(s.Substring(0, 2));
        var back = int.Parse(s.Substring(2, 2));
        var val = 1L * front * front + 1L * back * back;
        Console.WriteLine(val % 7 == 1 ? "YES" : "NO");
      }
    }
  }
}

C++

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

typedef long long ll;

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

  int T; cin >> T;
  while (T--) {
    string s; cin >> s;
    int front = stoi(s.substr(0, 2));
    int back = stoi(s.substr(2, 2));
    ll val = 1LL * front * front + 1LL * back * back;
    cout << (val % 7 == 1 ? "YES" : "NO") << "\n";
  }

  return 0;
}