FFT例题 模板hdu 1402 卷积 hdu 4609 3-idiots

贴几个FFT链接了，自己写真的写不来。。数论瞎子。。

 

https://blog.csdn.net/WADuan2/article/details/79529900

https://blog.csdn.net/ggn_2015/article/details/68922404

https://blog.csdn.net/f_zyj/article/details/76037583

 

hdu 1402

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1402

题意：两个最长为50000位的数的相乘，求积。

板题，就不说了直接上代码。

#include<bits/stdc++.h>
using namespace std;
const int maxn=200010;
typedef long long ll;
typedef unsigned long long ull;
// 这一大坨fft的代码实现不要动的
const double pi = acos(-1.0);
struct Complex {
    double x, y;
    Complex(double _x = 0.0, double _y = 0.0) {
        x = _x;
        y = _y;
    }
    Complex operator - (const Complex &b)const {
        return Complex(x-b.x, y-b.y);
    }
    Complex operator + (const Complex &b)const {
        return Complex(x+b.x, y+b.y);
    }
    Complex operator * (const Complex &b)const {
        return Complex(x*b.x-y*b.y, x*b.y+y*b.x);
    }
};

void change(Complex y[], int len) {
    int i, j, k;
    for(i = 1, j = len/2; i < len-1; i++) {
        if (i < j) swap(y[i], y[j]);
        k = len/2;
        while(j >= k) {
            j -= k;
            k /= 2;
        }
        if (j < k) j += k;
    }
}

void fft(Complex y[], int len, int on) {
    change(y, len);
    for(int h = 2; h <= len; h <<= 1) {
        Complex wn(cos(-on*2*pi/h), sin(-on*2*pi/h));
        for(int j = 0; j < len; j += h) {
            Complex w(1, 0);
            for(int k = j; k < j+h/2; k++) {
                Complex u = y[k];
                Complex t = w*y[k+h/2];
                y[k] = u+t;
                y[k+h/2] = u-t;
                w = w*wn;
            }
        }
    }
    if (on == -1)
        for(int i = 0; i < len; i++)
        y[i].x /= len;
}
char sa[maxn/2],sb[maxn/2];
Complex a[maxn],b[maxn],aim[maxn];
int ans[maxn];
int main(){
    while(~scanf("%s%s",sa,sb)){
        int n=strlen(sa),m=strlen(sb);
        int len=1,i;
        while(len<2*n||len<2*m) len<<=1;

        for(i=0;i<n;i++)
            a[i]=Complex(sa[n-i-1]-'0',0);
        for(;i<len;i++)
            a[i]=Complex(0,0);

        for(i=0;i<m;i++)
            b[i]=Complex(sb[m-i-1]-'0',0);
        for(;i<len;i++)
            b[i]=Complex(0,0);
        fft(a,len,1); fft(b,len,1);
        for(int i=0;i<len;i++){
            aim[i]=a[i]*b[i];
        }
        fft(aim,len,-1);
        for(int i=0;i<len;i++)
            ans[i]=(int)(aim[i].x+0.5);
        for(int i=0;i<len;i++){
            if(ans[i]>9){
                ans[i+1]+=ans[i]/10;
                ans[i]%=10;
            }
        }
        len--;
        while(ans[len]>9){
            ans[len+1]+=ans[len]/10;
            ans[len]%=10;
            len++;
        }
        while(ans[len]==0&&len>0) len--;
        for(int i=len;i>=0;i--)
            printf("%d",ans[i]);
        printf("\n");
    }
    return 0;
}

 

 

 

hdu 4609 题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4609

 

题意：给你n个数，从这n个数里随机取三个数，问着三个数能组成一个三角形的概率。

 

据说是FFT裸题，但说实话我乍一看真没看出来是FFT,知道是FFT也不会做...好不容易敲完代码发现保留7位而不是至少精确7位弄得我保留了9位..debug了两个多小时，果然还是我太弱了吗..

 

做法：

        算概率就有分子和分母，分母很明显就是，所以只要计算分子的数量。假设我们已经得到了对于一个数x，它是两个任意的数的和的种类数是sum[x]。

        这里有一个很神奇的操作，我们在加入答案的时候，枚举的是最大值，这样的话就可以避免重复，我们先把原数组进行排序，那么我们遍历的时候，要加上一个  的值，因为组成一个三角形要满足两边的和大于第三边，我们的sum数组存的就是两两组成的数字的和为x的种数。

        当然这里我们会发现一个问题，就是在相加的时候肯定会遇到重复，注意我们这里枚举的是最大值，所以不可能把还有大于这个数的情况加入，对于第i个数a[i]，这里的重复无非就是三种情况。(注：这里的下标i从0开始

        1.另外两个数都大于这个数，那么就要减掉(n-i-1)*(n-i-2)/2个数，（这里忽略到数字相同的情况，直接用下标来进行处理）。

        2.一个数为这第i个数，另一个数任意取，这样的方案数是n-1.

        3.一个数大于这个数，一个数小于这个数，这样的方案是(n-i-1)*i;

        减去这三种情况即可。

        该题的FFT用在处理sum数组上，多项式的表示为表示数字n出现的次数，用次数的数组num进行一个对自己的卷积，就可以得到sum数组，当然还要进行一次处理。对同一个数使用两次的情况要进行减法处理，对num数组还要除以2，不然一个情况会被重复计算一次。这样处理的复杂度就是  而不是.

---

代码如下，FFT还要加油。

#include<bits/stdc++.h>
using namespace std;
const int maxn=400040;
typedef long long ll;
const double pi = acos(-1.0);
struct Complex {
    double x, y;
    Complex(double _x = 0.0, double _y = 0.0) {
        x = _x;
        y = _y;
    }
    Complex operator - (const Complex &b)const {
        return Complex(x-b.x, y-b.y);
    }
    Complex operator + (const Complex &b)const {
        return Complex(x+b.x, y+b.y);
    }
    Complex operator * (const Complex &b)const {
        return Complex(x*b.x-y*b.y, x*b.y+y*b.x);
    }
};

void change(Complex y[], int len) {
    int i, j, k;
    for(i = 1, j = len/2; i < len-1; i++) {
        if (i < j) swap(y[i], y[j]);
        k = len/2;
        while(j >= k) {
            j -= k;
            k /= 2;
        }
        if (j < k) j += k;
    }
}

void fft(Complex y[], int len, int on) {
    change(y, len);
    for(int h = 2; h <= len; h <<= 1) {
        Complex wn(cos(-on*2*pi/h), sin(-on*2*pi/h));
        for(int j = 0; j < len; j += h) {
            Complex w(1, 0);
            for(int k = j; k < j+h/2; k++) {
                Complex u = y[k];
                Complex t = w*y[k+h/2];
                y[k] = u+t;
                y[k+h/2] = u-t;
                w = w*wn;
            }
        }
    }
    if (on == -1)
        for(int i = 0; i < len; i++)
        y[i].x /= len;
}
ll num[maxn],a[maxn/4];
ll sum[maxn];
Complex x[maxn];
int main(){
    int t,n;
    cin>>t;
    while(t--){
        memset(num,0,sizeof(num));
        scanf("%d",&n);
        for(int j=0;j<n;j++){
            scanf("%lld",&a[j]);
            num[a[j]]++;
        }
        sort(a,a+n);
        int len1=a[n-1]+1,len=1;
        while(len<2*len1) len<<=1;
        for(int i=0;i<len1;i++)
            x[i]=Complex(num[i],0);
        for(int i=len1;i<len;i++) x[i]=Complex(0,0);

        fft(x,len,1);
        for(int i=0;i<len;i++)
            x[i]=x[i]*x[i];
        fft(x,len,-1);

        for(int i=0;i<len;i++) num[i]=(ll)(x[i].x+0.5);

        len=2*a[n-1];
        for(int i=0;i<n;i++) num[2*a[i]]--;
        for(int i=0;i<=len;i++) num[i]/=2;
        for(int i=1;i<=len;i++)
            sum[i]=sum[i-1]+num[i];
        ll ans=0;
        for(int i=0;i<n;i++){
            //加上两项和比a[i]大的
            ans+=sum[len]-sum[a[i]];
            //减掉两项都比a[i]大的情况
            ans-=(ll)(n-i-1)*(n-i-2)/2;
            //减掉一个取自己,另一个任取的情况
            ans-=(n-1);
            //减掉一个取比自己大,一个比自己小的情况
            ans-=(ll)(n-i-1)*i;
        }
        ll fenmu=(ll)1ll*n*(n-1)*(n-2)/6;
        printf("%.7f\n",(double)1.0*ans/fenmu);
    }
    return 0;
}