一道算法题

问题描述：两个数组a[N]，b[N]，其中A[N]的各个元素值已知，现给b[i]赋值，b[i] = a[0]*a[1]*a[2]...*a[N-1]/a[i]；
要求：
1.不准用除法运算
2.除了循环计数值，a[N],b[N]外，不准再用其他任何变量（包括局部变量，全局变量等）

3.满足时间复杂度O（n），空间复杂度O（1）

 

思路：这题本来不是很难，但是条件3要求了时空复杂度。由于数组b其实是要求的。所以我们可以使用数组b的空间。b[i] = a[0]*a[1]*a[2]...*a[N-1]/a[i] => b[i]=(a[0]*···a[i-1])*(a[i+1]*···a[N-1])。上面的表达式已经很清楚了。我们可以把原来的连乘分为两段看。将数组b用来存放后半段连乘积。b[N-1]=1;b[i]=b[i+1]*a[i+1](i=N-2···0)。而前半段连乘我们可以用一个变量tmp来存储，这样其实用到的额外空间除了数组b就是tmp。空间复杂度O（1）满足。计算后半段连乘，一次遍历复杂度O（n）。构造要求的数组b。再一次遍历复杂度O（n）。合起来是O（2n）。

 

#include<stdio.h>
#include<iostream>

using namespace std;

void getb(int a[],int b[],int len)
{
        b[len-1]=1;
        for(int i=len-2;i>=0;i--)
                b[i]=b[i+1]*a[i+1];
        int tmp=1;
        for(int i=0;i<len;i++)
        {
                b[i]=b[i]*tmp;
                tmp=tmp*a[i];
        }
}

int main()
{
        int a[]={1,2,1,2,1,3,5,2};
        int len=sizeof(a)/sizeof(int);
        int* b=new int[len];
        getb(a,b,len);
        for(int i=0;i<len;i++)
                cout<<b[i]<<" ";
        cout<<endl;  
        return 0;
}