ArrayList删除第一个元素

Question

这是给定的问题:给定一个非负数表示为数字数组,

将数字加1(增加数字所代表的数字).

存储数字使得最高有效数字位于列表的开头.

例:

如果向量有[1,2,3]

返回的向量应为[1,2,4]

因为123 + 1 = 124.

这是我的代码:

 public class Solution {
    public ArrayList<Integer> plusOne(ArrayList<Integer> A) {       
        int carry = 1;
        int length = A.size();
        ArrayList result = new ArrayList();

        for( int i = length - 1; i >=0; i-- ){
            int val = A.get(i) + carry;
            result.add(0,val % 10);
            carry = val / 10;
        }

        if (carry == 1){
            result.add(0,1);
        }

        for (int j = 0; j < result.size(); j++){
            if(result.get(j).equals(0))
                result.remove(j);
            else
                break;
       }

        return result;

    }
  }

但是,在测试用例中:A:[0,6,0,6,4,8,8,1]

它说我的函数返回

6 6 4 8 8 2

而正确的答案是

6 0 6 4 8 8 2

我不知道我的代码有什么问题.

谢谢!

Answer 1

if(result.get(j).equals(0))
    result.remove(j);
else
    break;

如果每个其他索引包含0,这将失败.这是发生的事情:

0 6 0 6 4 8 8 2
^ (j = 0)

0将被删除,并j递增1.

6 0 6 4 8 8 2
  ^ (j = 1)

然后删除此0,跳过数组中的前6个.要解决此问题,请将代码段更改为:

if(result.get(j).equals(0))
    result.remove(j--);
else
    break;

这可以补偿何时删除索引,以便j在删除任何0后不会立即跳过该数字.