[Baekjoon] 1985_디지털 친구

Silver IV

문제(출처: https://www.acmicpc.net/problem/1985)

---

< 디지털 친구 >

 

문제 풀이 

 

x를 규칙에 따라 고친 후 y와 이루어져 있는 숫자가 일치하는지 확인하고, 일치하지 않다면 y를 규칙에 따라 고친 후 x와 이루어져 있는 숫자가 일치하는지 확인한다. 

 

 my solution (Java)

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;

public class _1985_ { // 디지털 친구
	static Set<Integer> set1, set2;

	public static void main(String[] args) throws IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st;

		for (int i = 0; i < 3; i++) {
			st = new StringTokenizer(bf.readLine());

			String x = st.nextToken();
			String y = st.nextToken();

			set1 = new HashSet<>();
			set2 = new HashSet<>();

			for (int j = 0; j < x.length(); j++) {
				set1.add(x.charAt(j) - '0');
			}
			for (int j = 0; j < y.length(); j++) {
				set2.add(y.charAt(j) - '0');
			}

			boolean flag = false;
			for (int num : set1) {
				if (set1.size() != set2.size() || !set2.contains(num)) {
					flag = true;
					break;
				}
			}
			if (!flag) {
				bw.write("friends\n");
			} else {
				if (change(x, true) || change(y, false)) {
					bw.write("almost friends\n");
				} else {
					bw.write("nothing\n");
				}
			}
		}
		bw.flush();
	}

	private static boolean change(String num, boolean flag) {
		char arr[] = num.toCharArray();
		for (int i = 0; i < num.length() - 1; i++) { // + -
			Set<Integer> answer = new HashSet<>();

			if (arr[i] - '0' + 1 > 9 || arr[i + 1] - '0' - 1 < 0) {
				continue;
			}
			answer.add(arr[i] - '0' + 1);
			answer.add(arr[i + 1] - '0' - 1);

			for (int j = 0; j < num.length(); j++) {
				if (j == i || j == i + 1) {
					continue;
				}
				answer.add(arr[j] - '0');
			}
			boolean check = false;
			for (int x : answer) {
				if (flag) {
					if (answer.size() != set2.size() || !set2.contains(x)) {
						check = true;
						break;
					}
				} else {
					if (answer.size() != set1.size() || !set1.contains(x)) {
						check = true;
						break;
					}
				}
			}
			if (!check) {
				return true;
			}
		}

		for (int i = 0; i < num.length() - 1; i++) { // - +
			Set<Integer> answer = new HashSet<>();

			if (arr[i] - '0' - 1 < 0 || arr[i + 1] - '0' + 1 > 9 || (i == 0 && arr[i] - '0' - 1 == 0)) {
				continue;
			}
			answer.add(arr[i] - '0' - 1);
			answer.add(arr[i + 1] - '0' + 1);

			for (int j = 0; j < num.length(); j++) {
				if (j == i || j == i + 1) {
					continue;
				}
				answer.add(arr[j] - '0');
			}
			boolean check = false;
			for (int x : answer) {
				if (flag) {
					if (answer.size() != set2.size() || !set2.contains(x)) {
						check = true;
						break;
					}
				} else {
					if (answer.size() != set1.size() || !set1.contains(x)) {
						check = true;
						break;
					}
				}
			}
			if (!check) {
				return true;
			}
		}
		return false;
	}
}

변수)
x, y : 두 정수 
set1 , set2 : x와 y가 이루어져 있는 숫자 구성 
flag : 이루어져 있는 숫자 구성 일치 여부
arr : 입력받은 숫자를 char 배열로 변환
answer : 규칙에 따라 바꾼 숫자가 이루어져 있는 숫자 구성 
check : 구성 일치 여부

 

정수 x와 y를 입력받는다. 각 x와 y를 구성하는 숫자를 set1과 set2에 저장한다. set1과 set2를 비교해서 일치한다면 "friends"를 출력한다. 일치하지 않다면 x와 y를 규칙에 맞게 고친 후 일치한다면 "almost friends"를 출력한다. 일치하지 않다면 "nothing"을 출력한다.

 

change(String num, boolean flag) 

num을 char 배열로 바꾼다. num을 순환하며 i에 1을 더하고 i+1에 1을 뺀다. 이때 0보다 작고 9보다 크다면 숫자 범위에 벗어나므로 종료한다. 바꾼 값을 구성하는 숫자를  answer에 저장한다. x를 규칙에 따라 바꾼 것이라면 answer과 set2를 비교한다. 일치하지 않다면 종료한다. y를 규칙에 따라 바꾼 것이라면 answer과 set1을 비교한다. 일치하지 않다면 종료하고 다음 규칙에 따라 num을 순환하며 i에 1을 빼고 i+1에 1을 더한다. 나머지는 위와 같이 구현한다.

---

---