【Leetcode】401. Binary Watch

5495 ワード

タイトルリンク:https://leetcode.com/contest/5/problems/binary-watch/
タイトル:A binary watch has 4 LEDs on the top which represent the hours(0-11)、and the 6 LEDs on the bottom represent the minutes(0-59).
Each LED represents a zero or one, with the least significant bit on the right.
For example, the above binary watch reads “3:25”.
Given a non-negative integer n which represents the number of LEDs that are currently on, return all possible times the watch could represent.
Example:
Input: n = 1 Return: [“1:00”, “2:00”, “4:00”, “8:00”, “0:01”, “0:02”, “0:04”, “0:08”, “0:16”, “0:32”] Note: The order of output does not matter. The hour must not contain a leading zero, for example “01:00” is not valid, it should be “1:00”. The minute must be consist of two digits and may contain a leading zero, for example “10:2” is not valid, it should be “10:02”.
考え方:時間、分の2つの部分をそれぞれ遡及し、結果を組み合わせます.
アルゴリズム:
    public List<String> readBinaryWatch(int k) {
        List<String> res = new ArrayList<String>();

        for (int h = 0; h <= k; h++) {
            dspCombination(hour, h, 0, true);
            dspCombination(minu, k - h, 0, false);

            for (int i : hs) {
                for (int j : ms) {
                    if (j == 0)
                        res.add("" + i + ":00");
                    else {
                        if (j / 10 == 0)
                            res.add("" + i + ":0" + j);
                        else {
                            res.add("" + i + ":" + j);
                        }
                    }
                }
            }

            hs.clear();
            ms.clear();
        }

        return res;
    }

    int hour[] = new int[] { 1, 2, 4, 8 };
    int minu[] = new int[] { 1, 2, 4, 8, 16, 32 };
    List<Integer> list = new ArrayList<Integer>();
    List<Integer> hs = new ArrayList<Integer>();
    List<Integer> ms = new ArrayList<Integer>();

    private void dspCombination(int[] nums, int k, int level, boolean flag) {
        if (list.size() == k) {
            int sum = 0;
            for (int num : list) {
                sum += num;
            }
            if (flag) {//    hour
                if (sum <= 11)
                    hs.add(sum);
            } else {
                if (sum <= 59)
                    ms.add(sum);
            }
            return;
        } else if (list.size() > k) {
            return;
        } else {
            for (int i = level; i < nums.length; i++) {
                list.add(nums[i]);
                dspCombination(nums, k, i + 1, flag);
                list.remove(list.size() - 1);
            }
        }
    }