(Java) 프로그래머스 나머지가 1이 되는 수 찾기

TreeSet을 이용하여 구현했다.

풀고나니 굳이 TreeSet을 이용해야 할 필요가 없다는 생각이 들었다....

 

---

최종코드

import java.util.TreeSet;
class Solution {
    public int solution(int n) {
        int answer = 0;
        // x를 넣을 treeset 초기화
        TreeSet<Integer> ts = new TreeSet<>();
        int x = 1;

        for (int i=0; i<n; i++) {
            if (n % x != 1) {
                x++;
            }else {
                ts.add(x);
                x++;
            }
        }

        answer = ts.pollFirst();
        return answer;
    }
}

 

다른 풀이

class Solution {
    public int solution(int n) {
        int answer = 0;
        for (int i=2; i<n; i++) {
            if (n % i == 1) {
                answer = i;
                break;
            } 
        }
        return answer;
    }
}