【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';
}