hdu4609 3idiots 三角形计数 FFT

直接上代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <algorithm>
using namespace std;
double pi = acos(-1.0);
struct complex{double re,im;complex(double r = 0.0,double i = 0.0):re(r),im(i){};complex operator+(complex com){return complex(re+com.re,im+com.im);}complex operator-(complex com){return complex(re-com.re,im-com.im);}complex operator*(complex com){return complex(re*com.re-im*com.im,re*com.im+im*com.re);}
};
complex wn,wntmp;
void rader(complex arr[],int n){int num = n-1;for(int i = 0;i < n;++i){int tn = n>>1;while(num && num >= tn) num ^= tn,tn >>= 1;num |= tn;if(num > i) swap(arr[i],arr[num]);}
}
void FFT(complex cs[],int n,int f){rader(cs,n);for(int s = 1;s < n;s <<= 1){wn = complex(cos(f*2*pi/(s*2)),sin(f*2*pi/(s*2)));for(int offset = 0;offset < n;offset += s<<1){wntmp = complex(1.0,0.0);for(int i = 0;i < s;++i){complex u = cs[offset+i],v = cs[offset+i+s]*wntmp;cs[offset+i] = u + v;cs[offset+i+s] = u - v;wntmp = wntmp * wn;}}}if(f == -1)for(int i = 0;i < n;++i)cs[i].re /= n;
}
const int maxn = 1e5+7;
int n;
long long ival[maxn<<2];
long long sum[maxn<<2];
complex val[maxn<<2];
long long ord[maxn<<2];
int main(){int T;scanf("%d",&T);while(T--){memset(val,0,sizeof(val));memset(ival,0,sizeof(ival));memset(sum,0,sizeof(sum));scanf("%d",&n);for(int i = 0;i < n;++i){int tmp;scanf("%d",&tmp);ord[i] = tmp;val[tmp].re++;ival[tmp] ++;}sort(ord,ord+n);int len = 1;while(len < ord[n-1]+1) len <<= 1;len <<= 1;FFT(val,len,1);for(int i = 0;i < len; ++i) val[i] = val[i]*val[i];FFT(val,len,-1);for(int i = 1;i < len+1;i++){if(i % 2 == 0) val[i].re = (val[i].re - ival[i/2]);val[i].re /= 2;}for(int i = 1;i < len+1;++i){ival[i] = (long long)(val[i].re+0.5);sum[i] = sum[i-1] + ival[i];}//sum[i]表示的是不重复，和<=i 的种类数//此时ival[i]表示的是两根不相同的棒棒组合在一起，长度和为i的种类数。long long ans = 0;for(int i = 0;i < n;++i){ans += sum[len] - sum[ord[i]] - (n-1) - (long long)(n-i-1)*(n-i-2)/2 - (long long)(i)*(n-i-1);}double fenmu = double(n)*(n-1)*(n-2)/6;printf("%.7lf\n",double(ans)/fenmu);}    return 0;
}

hdu4609 3idiots 三角形计数 FFT相关推荐

hdu4609 3-idiots
hdu4609 3-idiots 题意: 给出n个木棍以及它们的长度,求取其中三根长度各不相同的木棍组成三角形的概率题解: 三角形满足条件: 任意两边之和大于第三边我们设多项式A(x),其中Cix ...
nvGRAPH三角形计数和遍历示例
nvGRAPH三角形计数和遍历示例 #include " stdlib.h" #include" inttypes.h" #include" stdi ...
P2807 三角形计数（c++）
题目背景三角形计数(triangle) 递推题目描述把大三角形的每条边n等分,将对应的等分点连接起来(连接线分别平行于三条边),这样一共会有多少三角形呢?编程来解决这个问题. 输入格式第一行为 ...
69 三角形计数（Triangle Count）
文章目录 1 题目 2 解决方案 2.1 思路 2.2 时间复杂度 2.3 空间复杂度 3 源码 1 题目题目:三角形计数(Triangle Count) 描述:给定一个整数数组,在该数组中,寻找三 ...
[hdu4609]3-idiots（快速傅利叶变换FFT）
题目名字是三个制杖- [题意] 有T组数据每组数据给出n条线段,问任意取三条,可以组成三角形的概率 [输入] 开头一行输入T(T<=10) 下来T组数据,每组数据第一行输入一个n(3<= ...
3-idiots HDU - 4609 FFT模板
给N个木棍问任选三个可组合出三角形的概率以前写的,整理整理存个档 //#pragma comment(linker, "/STACK:1024000000,1024000000" ...
luogu2807三角形计数
题目描述把大三角形的每条边n等分,将对应的等分点连接起来(连接线分别平行于三条边),这样一共会有多少三角形呢?编程来解决这个问题. 输入输出格式输入格式: 第一行为整数t(≤100),表示测试数据 ...
FFT/NTT/FMT/FWT题目
快速傅里叶变换FFT hdu4609 3-idiots P4173 残缺的字符串 Hash Function [AH2017/HNOI2017]礼物 CF993E Nikita and Order S ...
spark的数三角形算法_数据算法 Hadoop Spark大数据处理技巧 PDF 下载
相关截图: 资料简介: 本书介绍了很多基本设计模式.优化技术和数据挖掘及机器学习解决方案,以解决生物信息学.基因组学.统计和社交网络分析等领域的很多问题.这本书还概要介绍了MapReduce.Hado ...

hdu4609 3idiots 三角形计数 FFT

hdu4609 3idiots 三角形计数 FFT相关推荐

最新文章

热门文章