방문 길이 (Summer/Winter Coding(~2018))

원본 : https://school.programmers.co.kr/learn/courses/30/lessons/49994

프로그래머스

 

풀이

import java.util.*;

class Solution {
    public int solution(String dirs) {
        Set<String> set = new HashSet<>();

        int preX;
        int preY;
        int nowX = 0;
        int nowY = 0;

        for (char dir : dirs.toCharArray()) {
            preX = nowX;
            preY = nowY;

            if (dir == 'U') {
                if (preY + 1 > 5 || preY + 1 < -5) {
                    continue;
                }
                nowY = preY + 1;
            } else if (dir == 'D') {
                if (preY - 1 > 5 || preY - 1 < -5) {
                    continue;
                }
                nowY = preY - 1;
            } else if (dir == 'L') {
                if (preX - 1 > 5 || preX - 1 < -5) {
                    continue;
                }
                nowX = preX - 1;
            } else if (dir == 'R') {
                if (preX + 1 > 5 || preX + 1 < -5) {
                    continue;
                }
                nowX = preX + 1;
            }

            set.addAll(Arrays.asList(getString(preX, preY, nowX, nowY)));
        }

        return set.size() / 2;
    }

    public String[] getString(int preX, int preY, int nowX, int nowY) {
        String[] result = new String[2];
        StringBuilder sb = new StringBuilder();

        sb.append("(").append(preX).append(", ").append(preY).append(")")
                .append(" -> ")
                .append("(").append(nowX).append(", ").append(nowY).append(")");

        result[0] = sb.toString();

        sb = new StringBuilder();

        sb.append("(").append(nowX).append(", ").append(nowY).append(")")
                .append(" -> ")
                .append("(").append(preX).append(", ").append(preY).append(")");

        result[1] = sb.toString();

        return result;
    }
}