• bzoj 3513: [MUTC2013]idiots【生成函数+FFT】


    想了好长时间最后发现真是石乐志
    第一反应就是两边之和大于第三边,但是这个东西必须要满足三次……
    任意的两边之和合通过生成函数套路+FFT求出来(记得去掉重复选取的),然后这任意两边之和大于任意第三边可以用一个前缀和求得(同样记得去重,前缀和里面一定包含前两条边),这样我们就得到了任意两边之和大于任意第三边的组数(这里是算顺序的,(1,2,3)(2,1,3)要算两遍)
    然后考虑任意选三条边方案数(算顺序),是( 6*C_n^3 ),注意到不符合要求的三条边一定是满足两次两边之和大于第三边的,所以(总方案数-任意两边之和大于任意第三边的组数)/2就是不合法的三条边方案数(不算顺序),然后也就能得到合法的三条边方案数(不算顺序)了
    然后除掉总方案数就是概率

    #include<iostream>
    #include<cstdio>
    #include<cmath>
    #include<cstring>
    using namespace std;
    const int N=300005;
    int T,n,m,q[N],bt,lm,re[N],s[N];
    long long ans;
    struct cd
    {
    	long double a,b;
    	cd(long double A=0,long double B=0)
    	{
    		a=A,b=B;
    	}
    	cd operator + (const cd &x) const
    	{
    		return cd(a+x.a,b+x.b);
    	}
    	cd operator - (const cd &x) const
    	{
    		return cd(a-x.a,b-x.b);
    	}
    	cd operator * (const cd &x) const
    	{
    		return cd(a*x.a-b*x.b,a*x.b+b*x.a);
    	}
    }a[N];
    int read()
    {
    	int r=0,f=1;
    	char p=getchar();
    	while(p>'9'||p<'0')
    	{
    		if(p=='-')
    			f=-1;
    		p=getchar();
    	}
    	while(p>='0'&&p<='9')
    	{
    		r=r*10+p-48;
    		p=getchar();
    	}
    	return r*f;
    }
    void dft(cd a[],int f)
    {
    	for(int i=0;i<lm;i++)
    		if(i<re[i])
    			swap(a[i],a[re[i]]);
    	for(int i=1;i<lm;i<<=1)
    	{
    		cd wi=cd(cos(M_PI/i),f*sin(M_PI/i));
    		for(int k=0;k<lm;k+=(i<<1))
    		{
    			cd w=cd(1,0),x,y;
    			for(int j=0;j<i;j++)
    			{
    				x=a[j+k],y=w*a[i+j+k];
    				a[j+k]=x+y,a[i+j+k]=x-y;
    				w=w*wi;
    			}
    		}
    	}
    	if(f==-1)
    		for(int i=0;i<lm;i++)
    			a[i].a/=lm;
    }
    int main()
    {
    	T=read();
    	while(T--)
    	{
    		memset(s,0,sizeof(s));
    		memset(a,0,sizeof(a));
    		m=0;
    		n=read();
    		for(int i=1;i<=n;i++)
    			q[i]=read(),s[q[i]]++,a[q[i]].a+=1,m=max(m,q[i]);
    		for(int i=1;i<=m*2;i++)
    			s[i]+=s[i-1];
    		for(bt=0;(1<<bt)<=2*m;bt++);
    		lm=(1<<bt);
    		for(int i=0;i<lm;i++)
    			re[i]=(re[i>>1]>>1)|((i&1)<<(bt-1));
    		dft(a,1);
    		for(int i=0;i<lm;i++)
    			a[i]=a[i]*a[i];
    		dft(a,-1);
    		for(int i=1;i<=n;i++)
    			a[q[i]+q[i]].a-=1;
    		ans=1ll*n*(n-1)*(n-2);
    		for(int i=1;i<=2*m;i++)
    			ans-=(long long)(a[i].a+0.5)*(s[i-1]-2);//cerr<<ans<<endl;
    		ans=1ll*n*(n-1)*(n-2)/6-ans/2;//cerr<<ans<<endl;
    		printf("%.7Lf
    ",(long double)ans*6/(long double)n/(long double)(n-1)/(long double)(n-2));
    	}
    	return 0;
    }
    
  • 相关阅读:
    Python环境变量设置
    Java基础视频笔记(八):多线程编程
    Java基础视频笔记(六):本地文件操作
    设计延迟加载的“单例设计模式”
    Java数组去除重复元素
    博客园真不错,终于发现一个清净之处
    [Architecture Design] 系统边界设计
    [Chatter] 架构设计是做甚么
    [DCF] Devices Communication Foundation Architecture V4
    [WPF] DataTemplate Binding to Interface
  • 原文地址:https://www.cnblogs.com/lokiii/p/10039107.html
Copyright © 2020-2023  润新知