BZOJ2194: 快速傅立叶之二(NTT,卷积)

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 1776  Solved: 1055
[Submit][Status][Discuss]

Description

请计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ，并且有 n < = 10 ^ 5。 a,b中的元素均为小于等于100的非负整数。

Input

第一行一个整数N,接下来N行，第i+2..i+N-1行，每行两个数，依次表示a[i],b[i] (0 < = i < N)。

Output

输出N行，每行一个整数，第i行输出C[i-1]。

Sample Input

5
3 1
2 4
1 1
2 4
1 4

Sample Output

24
12
10
6
1

HINT

 

Source

题目中给的公式不好搞

我们按照套路，将$B$翻转一下

$$C(k) = \sum_0^n a_i * b_{n - 1 - i + k}$$

此时后面的式子就只与$k$有关了

设$$D(n - 1 + k) = \sum_0^n a_i * b_{n - 1 - i + k}$$

直接NTT

#include<cstdio>
#define swap(x,y) x ^= y, y ^= x, x ^= y
#define LL long long
using namespace std;
const int MAXN =  * 1e5 + ;
inline int read() {
    char c = getchar(); int x = , f = ;
    while(c < '' || c > '') {if(c == '-') f = -; c = getchar();}
    while(c >= '' && c <= '') x = x *  + c - '',c = getchar();
    return x * f;
}
const int P = , g = , gi = ;
int N;
int
LL a[MAXN], b[MAXN], r[MAXN];
LL fastpow(LL a, int p, int mod) {
    LL base = ;
    while(p) {
        if(p & ) base = (base * a) % mod;
        a = (a * a) % mod; p >>= ;
    }
    return base % mod;
}
LL NTT(LL *A, int type, int N, int mod) {
    for(int i = ; i < N; i++)
        if(i < r[i]) swap(A[i], A[r[i]]);
    for(int mid = ; mid < N; mid <<= ) {
        LL W = fastpow( (type == ) ? g : gi, (P - ) / (mid << ), mod );
        for(int j = ; j < N; j += (mid << )) {
            int w = ;
            for(int k = ; k < mid; k++, w = (w * W) % P) {
                LL x = A[j + k] % P, y = w * A[j + k + mid] % P;
                A[j + k] = (x + y) % P;
                A[j + k + mid] = (x - y + P) % P;
            }
        }
    }
    if(type == -) {
        LL inv = fastpow(N, mod - , mod);
        for(int i = ; i < N; i++)
            A[i] = (A[i] * inv) % mod;
    }
}
int main() {
    #ifdef WIN32
    freopen("a.in","r",stdin);
    #endif
    N = read();
    for(int i = ; i < N; i++)
        a[i] = read(), b[N - i] = read();
    int limit = , L = ;
    while(limit <= N + N) limit <<=, L++;
    for(int i = ; i < limit; i++) r[i] = (r[i >> ] >> ) | ((i & ) << (L - ));
    NTT(a, , limit, P); NTT(b, , limit, P);
    for(int i = ; i <  limit; i++) a[i] = (a[i] * b[i]) % P;
    NTT(a, -, limit, P);
    for(int i = ; i < N * ; i++)
        printf("%d\n",a[i] % P);
    return ;
}