程序员代码面试指南刷题--第八章.不包含本位置值的累乘数组

题目描述

给定一个数组arr，返回不包含本位置值的累乘数组
例如，arr=[2,3,1,4]，返回[12, 8, 24, 6]，即除自己外，其他位置上的累乘
[要求]
时间复杂度为O(n)，额外空间复杂度为O(1)
输入描述:

第一行有两个整数N, P。分别表示序列长度，模数(即输出的每个数需要对此取模)
接下来一行N个整数表示数组内的数

输出描述:

输出N个整数表示答案

示例1
输入
4 100000007
2 3 1 4

输出
12 8 24 6

解法一：越界用long没得办法

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main{
    public static void product(int n,int p,int[] arr){
        long[] res =new long[n];
        res[0] =arr[0];
        for(int i=1;i<n;i++){
            res[i]=res[i-1]*arr[i]%p;
        }
        long tmp=1;//越界全部改成long
        for(int i=n-1;i>0;i--){
            res[i]=res[i-1]*tmp%p;
            tmp=(tmp*arr[i])%p;
        }
        res[0]=tmp%p;
       for(int i=0;i<n;i++){
                System.out.print(res[i]+" ");
                    }
    }
 public static void main(String[] args) throws IOException{
        BufferedReader bf =new BufferedReader(new InputStreamReader(System.in));
        String s="";
        while((s=bf.readLine())!=null){
           String[] nm=s.split(" ");
           int n=Integer.valueOf(nm[0]);
           int p=Integer.valueOf(nm[1]);
            nm=bf.readLine().split(" ");
            int[] arr=new int[n];
            for(int i=0;i<n;i++){
                arr[i]=Integer.valueOf(nm[i]);
            }
            product(n,p,arr);
        }
    }
}

空间复杂度为O(1)的情况下只能用除法来做了