前缀积和后缀积

Description:

给出数列A1,A2,...,AN，并设

Bi = (A1 * A2 * A3 ... AN) / Ai mod (109 + 7)

现要求把所有的Bi 算出来

Input:

输入包含多组测试数据。对于每组数据，第1 行，1 个整数N(1 ≤N ≤100,000), 表示数列的长度。第2

行，N 个整数A1,A2,...,AN(1 ≤Ai≤109)，表示给出的数列。输入以一个 0 表示结尾。

Output:

对于每组数据，输出一行，N 个整数用空格分隔，表示算出的B1,B2,...,BN。

Sample Input:

3

1 2 3

0

Sample Output:

6

3

2

 

题意并不难懂

题目给的数据很大，如果一直乘下去，肯定会溢出。所以这里用了一个取余的分配律(a*b)%c=(a%c*b%c)%c

但是这只能得到A1*A2......An，因为取余在除法中没有分配率，所以这里就利用了前缀积和后缀积来解决这个问题。

即:假如n=2,Bi=(A1*A2*A3*A4*A5)%Ai => pre[i]=A1%1000000007 suff[i]=(A3*A4*A5)%1000000007 Bi=pre[i]*suff[i]

代码如下：

 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
using namespace std;
long long a[100010],pre[100010],suff[100010];
const int mod=1000000007;
int main()
{
    int n,i;
    while(scanf("%d",&n)&&n)
    {
        memset(a,0,sizeof(a));
        for(i=1;i<=n;i++)
            scanf("%lld",&a[i]);
        pre[0]=1;
        for(i=1;i<=n;i++)
            pre[i]=(pre[i-1]*a[i])%mod;
        suff[n+1]=1;
        for(i=n;i>=1;i--)
            suff[i]=(suff[i+1]*a[i])%mod;
        for(i=1;i<=n;i++)
            printf("%lld%c",(pre[i-1]*suff[i+1])%mod,i==n?'\n':' ');
    }
    return 0;
}