【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';
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章