2016 acm香港网络赛 A题 A+B Problem （FFT）

题意：

给你一堆数,然后求ai+aj=ak的组成的（i,j,k）对有多少个,并且保证i,j,k下标互不相同。

题解：

num[k]表示（ai,aj）=k的个数。然后将a[i]+a[i]的那种重复的去掉。

然后计算当a[i]==k时的（i,j,k）有多少种，很明显就是 num[ a[i] ]种,

注意有一种情况是有0的时候,因为有可能会自己加了0也等于a[i]。所以要统计一下0有多少个,减了0就可以了,注意(i,j)也是有顺序要求的,（i,j）和(j,i)是2个不同的,所以还要处理有0的时候要乘以2倍。
然后是因为有负数,我们要全部都加上50000变成正数就好了。

（其实就是HDU4609的变形）

AC代码：

#include
#include
#include
#include
typedef long long ll;
using namespace std;
const int N = 2e5+10;
const double pi = acos(-1.0);

char s1[N],s2[N];
int len,res[N];

struct Complex
{
    double r,i;
    Complex(double r=0,double i=0):r(r),i(i) {};
    Complex operator+(const Complex &rhs)
    {
        return Complex(r + rhs.r,i + rhs.i);
    }
    Complex operator-(const Complex &rhs)
    {
        return Complex(r - rhs.r,i - rhs.i);
    }
    Complex operator*(const Complex &rhs)
    {
        return Complex(r*rhs.r - i*rhs.i,i*rhs.r + r*rhs.i);
    }
} va[N],vb[N];

//雷德算法--倒位序  
void rader(Complex F[],int len) //len = 2^M,reverse F[i] with  F[j] j为i二进制反转
{
    int j = len >> 1;
    for(int i = 1;i < len - 1;++i)
    {
        if(i < j) swap(F[i],F[j]);  // reverse
        int k = len>>1;
        while(j>=k)
        {
            j -= k;
            k >>= 1;
        }
        if(j < k) j += k;
    }
}
//FFT实现
void FFT(Complex F[],int len,int t)
{
    rader(F,len);
    for(int h=2;h<=len;h<<=1) //分治后计算长度为h的DFT 
    {
        Complex wn(cos(-t*2*pi/h),sin(-t*2*pi/h)); //单位复根e^(2*PI/m)用欧拉公式展开
        for(int j=0;j