hdu 4609

原创 2015年07月08日 21:24:31

题目描述:

1e5个长度1e5以内的棍子,随便挑出3根不同的(长度可以相同)的棍,问组成三角形的概率

题解:

主要分两块:(1)fft可以快速算出两个棍子和是多少和几个的情况.通过减去单个的和/2后续处理可以得到.(2)有了a+b,并没有a-b.于是思维转换,不对a+b挑c,而是枚举c找a+b.并且定义枚举的c是最大的(这里的最大,如果一样,则标号最大,只是为了不统计重复).这样巧妙地把a-b

重点:

关键是第二个思路的转换

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
#include <cmath>
#include <ctype.h>
#include <limits.h>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <queue>
#include <map>
#include <stack>
#include <set>
#include <bitset>
#define CLR(a) memset(a, 0, sizeof(a))
#define REP(i, a, b) for(ll i = a;i < b;i++)
#define REP_D(i, a, b) for(ll i = a;i <= b;i++)

typedef long long ll;

using namespace std;


const double PI = acos(-1.0);
const ll maxn = 5e5 + 100;
ll num[maxn], a[maxn];
ll sum[maxn];
ll n, len, num_n;


//复数结构体
struct Complex
{
    double x,y;//实部和虚部 x+yi
    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);
    }
};

Complex x1[maxn],x2[maxn];
/*
* 进行FFT和IFFT前的反转变换。
* 位置i和 (i二进制反转后位置)互换
* len必须去2的幂
*/
void change(Complex y[],ll len)
{
    ll i,j,k;
    for(i = 1, j = len/2; i <len-1; i++)
    {
        if(i < j)swap(y[i],y[j]);
//交换互为小标反转的元素,i<j保证交换一次
//i做正常的+1,j左反转类型的+1,始终保持i和j是反转的
        k = len/2;
        while(j >= k)
        {
            j -= k;
            k /= 2;
        }
        if(j < k)j += k;
    }
}
/*
* 做FFT
* len必须为2^k形式,
* on==1时是DFT,on==-1时是IDFT
*/
void fft(Complex y[],ll len,ll on)
{
    change(y,len);
    for(ll h = 2; h <= len; h <<= 1)
    {
        Complex wn(cos(-on*2*PI/h),sin(-on*2*PI/h));
        for(ll j = 0; j < len; j+=h)
        {
            Complex w(1,0);
            for(ll 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(ll i = 0; i < len; i++)
            y[i].x /= len;
}


void doFFT(ll a[], ll &len)
{
    ll len1 = 1;
    while(len1 < 2*len)
    {
        len1*=2;
    }
    for(ll i=0;i<len;i++)
    {
        x1[i] = Complex(a[i], 0);
    }
    for(ll i=len;i<len1;i++)
    {
        x1[i] = Complex(0, 0);
    }
    fft(x1, len1, 1);
    for(ll i=0;i < len1;i++)
    {
        x1[i] = x1[i]*x1[i];
    }
    fft(x1, len1, -1);
    len=len1;
    for(ll i=0;i < len;i++)
    {
        a[i] = (ll)(x1[i].x+0.5);
    }
}



ll select(ll x)
{
    if(x==0)
        return 0;
    if(x==1)
        return 0;
    return x*(x-1)/2;
}

void solve()
{
    sort(a, a+n);
    len = a[n-1]+1;
    num_n = len;
    CLR(num);
    REP(i, 0, n)
    {
        num[a[i]]++;
    }
    //fft
    doFFT(num, num_n);
    REP(i, 0, n)
    {
        num[a[i]+a[i]]--;
    }
    REP(i, 0, num_n)
    {
//        if(num[i]%2!=0)
//            printf("dfasdfasd");
        num[i] /= 2;
    }
    //printf("%I64d\n", num_n);
    sum[0]=0;
    REP(i, 1, num_n)
    {
        sum[i]=sum[i-1]+num[i];
    }
    ll ans=0;
    REP(i, 0, n)
    {
        ans += (sum[num_n-1] - sum[a[i]]);
        ans -= (n-1);
        ans -= select(n-i-1);
        ans -= (n-i-1)*i;
    }
    ll tot = n*(n-1)*(n-2)/6;
    double res = ans*1.0/(double)tot;
    printf("%.7f\n", res);
}

int main()
{
    //freopen("3Cin.txt", "r", stdin);
    //freopen("3Cout.txt", "w", stdout);
    ll t;
    scanf("%I64d", &t);
    while(t--)
    {
        scanf("%I64d", &n);
        REP(i, 0, n)
        {
            scanf("%I64d", &a[i]);
        }
        solve();
    }
    return 0;
}

HDU 4609 快速傅里叶变换

题意:给出10^5个边长,问任取三个组成三角形的概率。 这题没遇到还不知道什么是傅里叶变换FFT,当时怎么想时间复杂度都不能降到n*log(n)。实际是这样的,把每个记录长度个数的数组存在num数组...
  • h6363817
  • h6363817
  • 2013年07月25日 20:17
  • 1101

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

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

【并查集专题】【HDU】

How Many Tables Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) ...
  • zy691357966
  • zy691357966
  • 2014年11月29日 18:34
  • 1931

HDU 4609 快速傅里叶变换

题目链接题意:已知n条边的长度,任取三条不同的边能够组成三角形的概率。很经典的FFT 如果已知任意两条边的和的情况,那么枚举第三条边可以在O(n)O(n)的复杂度下得出总方案数。难点便在于如果枚举两...
  • WuBaizhe
  • WuBaizhe
  • 2017年08月03日 10:23
  • 152

【HDU】4609 3-idiots 【FFT】

传送门:【HDU】4609 3-idiots
  • u013368721
  • u013368721
  • 2015年04月28日 18:01
  • 684

HDU 4609 3-idiots

FFT+容斥原理+神奇的思路
  • SenyeLicone
  • SenyeLicone
  • 2017年03月08日 16:45
  • 124

HDU 4609 FFT+各种分类讨论

思路: http://www.cnblogs.com/kuangbin/archive/2013/07/24/3210565.html 其实我是懒得写了....一定要define int long...
  • qq_31785871
  • qq_31785871
  • 2017年01月25日 00:27
  • 598

HDU 4609 3-idiots(FFT)

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

HDU 4609 3-idiots FFT+计数问题

题目大意: 就是现在给出n个长度不超过100000的树枝 的长度, n 大致思路: 这个题还是很好的一道计数问题, 首先由于n 那么对于一个三角形, 考虑其最大的边, 首先将n根树...
  • u013738743
  • u013738743
  • 2015年07月15日 12:32
  • 1208

hdu 4609 3-idiots 快速傅里叶变换/FFT

#include #include #include #include #include #include using namespace std; #define maxn 400040...
  • a601025382s
  • a601025382s
  • 2013年08月17日 17:30
  • 2333
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:hdu 4609
举报原因:
原因补充:

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