hdu 4609 3-idiots (FFT)

原创 2016年06月01日 08:39:13

被每个长度生成一个函数,个数作为系数,然后FFT一下,然后把每条边作为最长边算一下,注意减去他不是最长边的情况。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=220005;
const double eps(1e-8);
typedef long long LL;
const double PI = acos(-1.0);

struct Complex
{
    double real, image;
    Complex(double _real, double _image)
    {
        real = _real;
        image = _image;
    }
    Complex(){}
};

Complex operator + (const Complex &c1, const Complex &c2)
{
    return Complex(c1.real + c2.real, c1.image + c2.image);
}

Complex operator - (const Complex &c1, const Complex &c2)
{
    return Complex(c1.real - c2.real, c1.image - c2.image);
}

Complex operator * (const Complex &c1, const Complex &c2)
{
    return Complex(c1.real*c2.real - c1.image*c2.image, c1.real*c2.image + c1.image*c2.real);
}

int rev(int id, int len)
{
    int ret = 0;
    for(int i = 0; (1 << i) < len; i++)
    {
        ret <<= 1;
        if(id & (1 << i)) ret |= 1;
    }
    return ret;
}

Complex A[maxn<<1];
void FFT(Complex* a, int len, int DFT)//对a进行DFT或者逆DFT, 结果存在a当中,len必须是2的幂而且len大于多项式最高次数
{
    for(int i = 0; i < len; i++)
        A[rev(i, len)] = a[i];
    for(int s = 1; (1 << s) <= len; s++)
    {
        int m = (1 << s);
        Complex wm = Complex(cos(DFT*2*PI/m), sin(DFT*2*PI/m));
        for(int k = 0; k < len; k += m)
        {
            Complex w = Complex(1, 0);
            for(int j = 0; j < (m >> 1); j++)
            {
                Complex t = w*A[k + j + (m >> 1)];
                Complex u = A[k + j];
                A[k + j] = u + t;
                A[k + j + (m >> 1)] = u - t;
                w = w*wm;
            }
        }
    }
    if(DFT == -1) for(int i = 0; i < len; i++) A[i].real /= len, A[i].image /= len;
    for(int i = 0; i < len; i++) a[i] = A[i];
    return;
}
Complex a[maxn<<1];
int san[maxn];
int num[maxn<<1];
LL xishu[maxn<<1];
int main()
{
    int t,n;
    cin>>t;
    while(t--){
        scanf("%d",&n);
        int len=1;
        int s=0;
        for(int i=1;i<=n;i++){
            scanf("%d",&san[i]);
            s=max(s,san[i]);
        }
        while(len<=s){
            len<<=1;
        }
        len<<=1;
        sort(san+1,san+1+n);
        for(int i=0;i<len;i++){
            a[i]=Complex(0,0);
            num[i]=0;
        }
        for(int i=1;i<=n;i++){
            a[san[i]].real+=1;
            num[san[i]]++;
        }
        FFT(a,len,1);
        for(int i=0;i<len;i++){
            a[i]=a[i]*a[i];
        }
        FFT(a,len,-1);
        for(int i=0;i<len;i++){
            xishu[i]=(LL)(a[i].real+0.5);
        }
        for(int i=0;i<len/2;i++){
            xishu[2*i]-=num[i];
        }
        for(int i=0;i<len;i++){
            xishu[i]/=2;
        }
        xishu[len]=0;
        for(int i=len-1;i>=0;i--){
            xishu[i]+=xishu[i+1];
        }
        LL ans=0;
        for(int i=n;i>=1;i--){
            ans+=xishu[san[i]+1];
            ans-=((LL)(i+n-2)*(n-i+1))/2;
        }
        double ans1=1.0*ans/((LL)n*(n-1)*(n-2)/6);
        printf("%.7lf\n",ans1);
    }
}


相关文章推荐

HDU 4609 3-idiots(FFT学习)

附上kuangbin大神博客:讲解的非常清楚分析:暴力枚举aka_k,那么问题变成了找满足ai+aj>ak(i,j,k互不相等)a_i+a_j>a_k(i,j,k互不相等)的(ai,aj)(a_i,a...

HDU 4609 3-idiots [FFT]

HDU 4609 3-idiots [FFT]

hdu 4609 3-idiots 【FFT快速傅立叶变换】

FFT实现起来挺复杂的,开始用vector,结果发现空间超了,换成数组还是超,删掉了几个后又超时了 sin cos 函数调用次数太多了,改成乘法,还是超时 最后把FFT里的除法运算和模运算优化了一下,...
  • zz_1215
  • zz_1215
  • 2014年08月01日 19:56
  • 1046

hdu 4609 3-idiots [FFT计数]【数学】

题目连接:http://acm.hdu.edu.cn/showproblem.php?pid=4609 ———————————————————–. 3-idiotsTime Limit: 1000...

HDU 4609 3-idiots(FFT优化 + 计数方法)——2013 Multi-University Training Contest 1

传送门 3-idiotsTime Limit: 10000/5000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)Tot...

【HDU4609】3-idiots(FFT+计数问题)

题目链接 题目大意: 给你n个边长,问能组成三角形的概率如果用一个多项式A(x), 其x^k的系数ak表示长度为k的树枝的数量, 那么A(x)的平方当中x^k的系数就是从这些树枝中取两根可重复的和...

HDU 4609 3-idiots三个智障 FFT+组合计数

HDU4609 这道三个智障的题目把我做成了智障,先膜拜bin神,他的题解已经很详细 Bin神的HDU4609题解 这里稍作一些解释: num[i]num[i]数组记录的不同长度的strick...

【HDU】4609 3-idiots 【FFT】

传送门:【HDU】4609 3-idiots

HDU 4609 3-idiots(FFT)

题意:给定n(n 思路:这道题实际上就是问能组成多少三角形,如果直接枚举的话时间复杂度为O(n*n),无法承受。 注意到每根火柴棍的长度都小于等于10^5,所以可以把火柴棍的长度写成多项式的形式,...

HDU 4609 3-idiots FFT+计数问题

题目大意: 就是现在给出n个长度不超过100000的树枝 的长度, n 大致思路: 这个题还是很好的一道计数问题, 首先由于n 那么对于一个三角形, 考虑其最大的边, 首先将n根树...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:hdu 4609 3-idiots (FFT)
举报原因:
原因补充:

(最多只允许输入30个字)