有一个问题，其中给出了两个随机整数数组，其中从 0 到 9 的数字出现在每个索引处(即单个数字整数出现在两个给定数组的每个索引处)。我需要找到输入数组表示的数字的总和并将结果放在另一个数组中。

我相信我的代码一切正常，因为我为不同的数组执行了将近 50 到 60 次。但是当我在学校的在线法官中提交它时，它只接受了 4 个测试用例并拒绝了另外两个。我不知道在哪种情况下它会给出错误的输出。需要一点帮助。

这是我的代码

public static int[] sumOfTwoArrays(int[] arr1, int[] arr2){
    int size1 = arr1.length;
    int size2 = arr2.length;
    int carry = 0,sum,s,r;
    if(size1 == size2) {
        int arr3[] = new int[size1+1];
        for(int i=arr1.length-1;i>=-1;i--) { 
            if(i==-1) {
                arr3[i+1] = carry;
                //System.out.println(i+1+" "+arr3[i+1]);
            } else {
                sum = arr1[i] + arr2[i];
                if(sum>9) {
                    s =sum;
                    r = s % 10;
                    arr3[i+1] = carry + r;
                    carry = 1;
                    //System.out.println(i+" "+arr3[i]);    
                } else {
                    if(sum==9 && carry==1) {
                        s =sum+carry;
                        r = s % 10;
                        arr3[i+1] = r;
                    } else {
                        arr3[i+1] = sum+carry;
                        carry=0; 
                    }
                    //System.out.println(i+" "+arr3[i]);
                }  
            }      
        }
        return arr3;
    } else if (size1>size2) {
       int arr3[] = new int[size1+1];
       int diff = arr1.length - arr2.length;
       for(int i=arr1.length-1;i>=-1;i--) {
           if(i==-1) {
               arr3[i+1] = carry;
           } else {
               if(i>=diff) {
                   sum = arr1[i] + arr2[i-diff];
                    if(sum>9) {
                        s =sum;
                        r = s % 10;
                        arr3[i+1] = carry + r;
                        carry = 1;
                    } else {
                        if(sum==9 && carry==1) {
                            s =sum+carry;
                            r = s % 10;
                            arr3[i+1] = r;
                        } else {
                            arr3[i+1] = sum+carry;
                            carry=0; 
                        }
                    } 
                }  // end of diff i
                else {
                   arr3[i+1] =  arr1[i];
                   carry = 0;
                }
            }      
        }
        return arr3;
    } else {
        int arr3[] = new int[size2+1];
        int diff = arr2.length - arr1.length;
        for(int i=arr2.length-1;i>=-1;i--) {
            if(i==-1) {
                arr3[i+1] = carry;
            } else {
                if(i>=diff) {
                    sum = arr2[i] + arr1[i-diff];
                    if(sum>9) {
                        s =sum;
                        r = s % 10;
                        arr3[i+1] = carry + r;
                        carry = 1;
                    } else {
                        if(sum==9 && carry==1) {
                            s =sum+carry;
                            r = s % 10;
                            arr3[i+1] = r;
                        } else {
                            arr3[i+1] = sum+carry;
                            carry=0; 
                        }
                    }  
                }  // end of diff i
                else {
                    arr3[i+1] =  arr2[i];
                    carry = 0;
                }
            }      
        }
        return arr3;
    }   
}

示例输入:

int[] arr1 = {8,5,3,9,6};
int[] arr2 = {3,3,3,3,3};

示例输出:

{1,1,8,7,2,9}

示例输入:

int[] arr1 = {8,5,3,9,6};
int[] arr2 = {1,0,5}; 

示例输出:

{0,8,5,5,0,1}

最佳答案

好吧，我有这个基于 Eran 解决方案的算法(正在努力修复他纠正后的错误)，我会分享它，因为我使用的数组较少。

public static int[] sum(int[] arr1, int[] arr2){
    int carry = 0;
    int sum = 0;

    int len1 = arr1.length;
    int len2 = arr2.length;
    int len = Math.max(len1, len2);

    int arr3[] = new int[len + 1];

    for (int i = 1; i <= len; i++) {
        sum =
            (len1 - i >= 0 ? arr1[len1-i] : 0)
            + (len2 - i >= 0 ? arr2[len2-i] : 0)
            + carry;

        arr3[len-i+1] = sum%10;
        carry = sum/10;
    }
    arr3[0] = carry;

    return arr3;
}

三元运算符的用法仍然可读，所以我发现这是一个很好的解决方案。

为了简短说明，我们从末尾读取数组，使用 i 从右到左读取，但基于数组的长度。三元运算用于不同数组大小的情况。

编辑:

您的算法无法正确管理不同大小数组的进位值。

185 + 16 得到 101。

只是因为你设置了这样的值:

arr3[i+1] =  arr1[i];

所以你忘记了在最后一个操作中可能发生的进位。

关于java - 两个数组元素的总和？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/41973522/