문제
재용이는 최신 컴퓨터 10대를 가지고 있다. 어느 날 재용이는 많은 데이터를 처리해야 될 일이 생겨서 각 컴퓨터에 1번부터 10번까지의 번호를 부여하고, 10대의 컴퓨터가 다음과 같은 방법으로 데이터들을 처리하기로 하였다.
1번 데이터는 1번 컴퓨터, 2번 데이터는 2번 컴퓨터, 3번 데이터는 3번 컴퓨터, ... ,
10번 데이터는 10번 컴퓨터, 11번 데이터는 1번 컴퓨터, 12번 데이터는 2번 컴퓨터, ...
총 데이터의 개수는 항상 ab개의 형태로 주어진다. 재용이는 문득 마지막 데이터가 처리될 컴퓨터의 번호가 궁금해졌다. 이를 수행해주는 프로그램을 작성하라.
입력
입력의 첫 줄에는 테스트 케이스의 개수 T가 주어진다. 그 다음 줄부터 각각의 테스트 케이스에 대해 정수 a와 b가 주어진다. (1 ≤ a < 100, 1 ≤ b < 1,000,000)
출력
각 테스트 케이스에 대해 마지막 데이터가 처리되는 컴퓨터의 번호를 출력한다.
백준의 재채점으로 틀렸습니다가 떠서 다시 푼 문제다.
어디가 문제인가 했더니, b=1인 경우 틀린 답이 나온 것 같다.
아래 코드는 틀렸습니다가 뜬 1년 전에 푼 코드이다.
맨 아래 정답 코드와 비교해보면, b번의 곱을 하는 임시 변수 c의 초기값이 a였던게 문제이다.
이 경우 b가 1이면 c를 10으로 나누지 않게 되고 결국 잘못된 값이 나온다.
예를 들어 a=23, b=1 이라 해보자. c=23이 되고 b가 1이므로 10으로 나눗셈을 하지 않고 그대로 23을 출력하게 된다.
따라서 임시변수를 맨 처음에 1로 초기화 한 후, 정확히 b번만큼 a를 곱하고 10을 나누자.
#include <iostream>
#include <math.h>
using namespace std;
int main()
{
int T;
cin >> T;
for (int i = 1; i <= T; i++) {
int a, b;
cin >> a >> b;
int c = a;
for (int i = 1; i < b; i++) {
c *= a;
c %= 10;
}
if (c == 0) cout << 10 << endl;
else cout << c << endl;
}
}
C++ 코드
#include <iostream>
using namespace std;
int T, a, b;
int main()
{
cin >> T;
while (T--) {
cin >> a >> b;
int res = 1;
for (int i = 1; i <= b; i++) {
res = (res * a) % 10;
}
if (res == 0) cout << 10 << "\n";
else cout << res << "\n";
}
}
'알고리즘 > 백준' 카테고리의 다른 글
[백준] N-Queen (9663번) (0) | 2020.03.19 |
---|---|
[백준] 숫자판 점프 (2210번) (0) | 2020.03.19 |
[백준] 이모티콘 (14226번) (0) | 2020.03.14 |
[백준] 사전 (1256번) (0) | 2020.03.12 |
[백준] 색종이 만들기 (2630번) (0) | 2020.03.12 |