HDU4609 3-idiots

2021-07-28 23:34:25 阅读：204 来源： 互联网

标签：oth ch const HDU4609 int Comp idiots include

传送门

最近比赛遇到了要用FFT的题，就趁机加强一下。

一句话题意：在\(n\)个数中随便拿出三个数，问能组成三角形的概率。（\(1 \leqslant n \leqslant 10^ 5, 1 \leqslant a_i \leqslant10 ^ 5\)）

看到\(a_i\)的限制，就能想到开一个桶\(num[i]\)表示大小为\(i\)的数的个数。

那么\(f = num ^ 2\)，\(f(i)\)就表示拿两个数，组成大小为\(i\)的数的方案数。但这里面有不合法的和重复的，所以要将\(f(a_i+a_i)\)减\(1\)，以及将所有\(f(i)\)除以\(2\)，因为\(a_x+a_y\)和\(a_y+a_x\)是同一种选法。

上述的\(f(i)\)直接用FFT即可计算，接下来考虑怎么组成三角形。

我们可以枚举长边。
那么对于每一个长边\(a_i\)，可以和\(\sum\limits_{j>a_i} f(j)\)组成三角形，但这样算会多。

一是可能会有一个比\(a_i\)长的边，一个比\(a_i\)短的边，那么方案数就要减去\((i-1)*(n-i)\).
二是可能一个刚好也选\(a_i\)（注意，是选\(a_i\)本身，而选和\(a_i\)相同的数是合法的，而且两个短边不可能同时选到\(a_i\)，因为这种情况再刚求完\(f\)的时候已经去掉了），另一个随便选，因此要再减去\(n-1\).
最后一种就是选了两个都比\(a_i\)大的数，再减去\(C_{n - i}^2\).

总的来说这题难点不在于卷积，而是后面组合数的推导，要不重不漏的考虑到所有情况。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cctype>
#include<vector>
#include<queue>
#include<assert.h>
#include<ctime>
using namespace std;
#define enter puts("") 
#define space putchar(' ')
#define Mem(a, x) memset(a, x, sizeof(a))
#define In inline
#define forE(i, x, y) for(int i = head[x], y; ~i && (y = e[i].to); i = e[i].nxt)
typedef long long ll;
typedef double db;
const int INF = 0x3f3f3f3f;
const db eps = 1e-8;
const int maxn = 3e5 + 5;
const db PI = acos(-1);
In ll read()
{
	ll ans = 0;
	char ch = getchar(), las = ' ';
	while(!isdigit(ch)) las = ch, ch = getchar();
	while(isdigit(ch)) ans = (ans << 1) + (ans << 3) + ch - '0', ch = getchar();
	if(las == '-') ans = -ans;
	return ans;
}
In void write(ll x)
{
	if(x < 0) x = -x, putchar('-');
	if(x >= 10) write(x / 10);
	putchar(x % 10 + '0');
}

int n, a[maxn];

int len = 1, lim = 0, rev[maxn];
struct Comp
{
	db x, y;
	In Comp operator + (const Comp& oth)const {return (Comp){x + oth.x, y + oth.y};}
	In Comp operator - (const Comp& oth)const {return (Comp){x - oth.x, y - oth.y};}
	In Comp operator * (const Comp& oth)const {return (Comp){x * oth.x - y * oth.y, x * oth.y + y * oth.x};}
	friend In void swap(Comp &a, Comp& b) {swap(a.x, b.x), swap(a.y, b.y);}
}A[maxn];
In void fft(Comp* a, int flg)
{
	for(int i = 0; i < len; ++i) if(i < rev[i]) swap(a[i], a[rev[i]]);
	for(int i = 1; i < len; i <<= 1)
	{
		Comp omg = (Comp){cos(PI / i), sin(PI / i) * flg};
		for(int j = 0; j < len; j += (i << 1))
		{
			Comp o = (Comp){1, 0};
			for(int k = 0; k < i; ++k, o = o * omg)
			{
				Comp tp1 = a[k + j], tp2 = o * a[k + j + i];
				a[k + j] = tp1 + tp2, a[k + j + i] = tp1 - tp2;
			}
		}
	}
}

ll num[maxn], sum[maxn];
In ll solve()
{
	sort(a + 1, a + n + 1);
	for(int i = 1; i <= n; ++i) num[a[i]]++;
	int N = a[n] + a[n];
	while(len <= N) len <<= 1, ++lim;
	for(int i = 0; i < len; ++i) rev[i] = (rev[i >> 1] >> 1) | ((i & 1) << (lim - 1));
	for(int i = 0; i <= a[n]; ++i) A[i] = (Comp){num[i], 0};
	for(int i = a[n] + 1; i < len; ++i) A[i] = (Comp){0, 0};
	fft(A, 1);
	for(int i = 0; i < len; ++i) A[i] = A[i] * A[i];
	fft(A, -1);
	for(int i = 0; i <= N; ++i) num[i] = A[i].x / len + 0.5;
	for(int i = 1; i <= n; ++i) num[a[i] + a[i]]--;
	for(int i = 1; i <= N; ++i) num[i] /= 2, sum[i] = sum[i - 1] + num[i];
	ll ret = -1LL * n * (n - 1);
	for(int i = 1; i <= n; ++i)
	{
		ret += sum[N] - sum[a[i]];
		ret -= 1LL * (i - 1) * (n - i);
		ret -= (1LL * (n - i) * (n - i - 1) >> 1);
	}
	return ret;
}

In void init()
{
	Mem(num, 0);
	len = 1, lim = 0;
}

int main()
{
	int T = read();
	while(T--)
	{
		init();
		n = read();
		for(int i = 1; i <= n; ++i) a[i] = read();
		printf("%.7lf\n", 1.0 * solve() / (n - 2) / (n - 1) / n * 6);
	}
	return 0;
}

标签：oth,ch,const,HDU4609,int,Comp,idiots,include
来源： https://www.cnblogs.com/mrclr/p/15073151.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

HDU4609 3-idiots