bzoj3771 Triple

最新推荐文章于 2019-10-31 12:40:27 发布

sdfzyhx

最新推荐文章于 2019-10-31 12:40:27 发布

阅读量256

点赞数

分类专栏：数学 bzoj 文章标签： FFT

本文链接：https://blog.csdn.net/sdfzyhx/article/details/70245711

版权

bzoj 同时被 2 个专栏收录

251 篇文章 0 订阅

订阅专栏

数学

229 篇文章 0 订阅

订阅专栏

很明显是卷积的形式，但是要注意容斥。最后推出来答案是

a \times a \times a - 3 ( a \cdot a ) \times a + 2 a \cdot a \cdot a 6 + a \times a - a \cdot a 2 + a

$\frac{a\times a\times a-3(a\cdot a)\times a+2a\cdot a \cdot a}6+\frac{a\times a-a\cdot a}2+a$

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define LL long long
const double pi=acos(-1);
const int maxn=200010;
struct Complex
{
    double x,y;
    Complex operator + (const Complex &c) const
    {
        return (Complex){x+c.x,y+c.y};
    }
    Complex operator - (const Complex &c) const
    {
        return (Complex){x-c.x,y-c.y};
    }
    Complex operator * (const Complex &c) const
    {
        return (Complex){x*c.x-y*c.y,x*c.y+y*c.x};
    }
}a[maxn],in2[maxn],in3[maxn],conv2[maxn],conv3[maxn],w[maxn],t1,t2;
int m,l,rev[maxn];
LL ans1[maxn],ans2[maxn],ans3[maxn];
void fft(Complex *a,int fl)
{
    int x;
    for (int i=0;i<l;i++)
        if (rev[i]>i) swap(a[i],a[rev[i]]);
    for (int i=1;i<=m;i++)
        for (int j=0;j<l;j+=1<<i)
        {
            x=0;
            for (int k=j;k<j+(1<<i-1);k++)
            {
                t1=a[k];
                t2=a[k+(1<<i-1)];
                a[k]=t1+t2*w[x];
                a[k+(1<<i-1)]=t1-t2*w[x];
                x+=fl*(1<<m-i);
                if (x<0) x+=l;
            }
        }
}
int main()
{
    int n,mm=0,x;
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        mm=max(mm,x);
        a[x].x++;
    }
    while ((1<<m)-1<3*mm) m++;
    l=1<<m;
    for (int i=0;i<l;i++)
        for (int j=0;j<m;j++)
            rev[i]|=((i>>j)&1)<<m-j-1;
    w[0]=(Complex){1,0};
    w[1]=(Complex){cos(2*pi/l),sin(2*pi/l)};
    for (int i=2;i<l;i++)
    {
        w[i]=w[i/2]*w[i/2];
        if (i&1) w[i]=w[i]*w[1];
    }
    for (int i=0;i<=mm;i++)
    {
        in2[i*2]=a[i]*a[i];
        in3[i*3]=a[i]*a[i]*a[i];
    }
    for (int i=0;i<l;i++)
    {
        ans1[i]=(LL)(a[i].x+0.5);
        ans2[i]=-(LL)(in2[i].x+0.5);
        ans3[i]=(LL)2*(in3[i].x+0.5);
    }
    fft(a,1);
    for (int i=0;i<l;i++)
    {
        conv2[i]=a[i]*a[i];
        conv3[i]=a[i]*a[i]*a[i];
    }
    fft(conv2,-1);
    for (int i=0;i<l;i++) ans2[i]+=(LL)(conv2[i].x/l+0.5);
    fft(conv3,-1);
    for (int i=0;i<l;i++) ans3[i]+=(LL)(conv3[i].x/l+0.5);
    fft(in2,1);
    for (int i=0;i<l;i++) in2[i]=in2[i]*a[i];
    fft(in2,-1);
    for (int i=0;i<l;i++) ans3[i]-=3*(LL)(in2[i].x/l+0.5);
    for (int i=0;i<l;i++)
        if (ans3[i]/6+ans2[i]/2+ans1[i])
            printf("%d %lld\n",i,ans3[i]/6+ans2[i]/2+ans1[i]);
}