【LeetCode】1404-將二進制表示減到 1 的步驟數

題目描述

給你一個以二進制形式表示的數字 s 。請你返回按下述規則將其減少到 1 所需要的步驟數:

如果當前數字爲偶數,則將其除以 2 。

如果當前數字爲奇數,則將其加上 1 。

題目保證你總是可以按上述規則將測試用例變爲 1 。

示例 1:

輸入:s = "1101"
輸出:6
解釋:“1101” 表示十進制數 13 。
Step 1) 13 是奇數,加 1 得到 14
Step 2) 14 是偶數,除 2 得到 7
Step 3) 7 是奇數,加 1 得到 8
Step 4) 8 是偶數,除 2 得到 4
Step 5) 4 是偶數,除 2 得到 2
Step 6) 2 是偶數,除 2 得到 1

示例 2:

輸入:s = "10"
輸出:1
解釋:“10” 表示十進制數 2 。
Step 1) 2 是偶數,除 2 得到 1

示例 3:

輸入:s = "1"
輸出:0

提示:

  • 1 <= s.length <= 500
  • s 由字符 '0''1' 組成。
  • s[0] == '1'

Solution

這一題主要是根據題目意思來模擬處理,首先編寫check函數判斷字符數組是否滿足前面的位數都是0,最後一位是1;其次根據當前末位是1還是0來判斷當前數是奇數還是偶數,末位是1爲奇數,需要加1,需要注意的是首位的進位要增加數組長度;末位爲0是偶數,只需要右移一位即可。

public int numSteps(String s) {
    char[] ch = s.toCharArray();
    int count = 0;
    while (!check(ch)) {
        int n = ch.length;
        if (ch[n - 1] == '1') { // 奇數
            ch[n - 1] = '2'; // 末尾加一變成 2
            for (int i = n - 2; i >= 0; i--) { // 遍歷進位
                if (ch[i + 1] == '2') {
                    ch[i + 1] = '0';
                    ch[i]++;
                }
            }
            // 如果首位爲 2,說明需要增加位數
            if (ch[0] == '2') {
                ch[0] = '0';
                char[] temp = new char[n + 1];
                System.arraycopy(ch, 0, temp, 1, ch.length);
                ch = temp;
                temp[0] = '1';
            }
        } else { // 偶數:右移一位(相當於除以 2 )
            for (int i = n - 1; i > 0; i--) {
                ch[i] = ch[i - 1];
            }
            ch[0] = '0';
        }
        count++;
    }
    return count;
}

/**
 * 判斷是否爲 1
 * @param ch 字符串數組
 * @return
 */
public boolean check(char[] ch) {
    for (int i = 0; i < ch.length - 1; i++) {
        if (ch[i] != '0')
            return false;
    }
    return ch[ch.length - 1] == '1';
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章