FFT+简单生成函数+容斥原理两题

一.HDU4609

题目大意:

给你一个序列,任意选出三个不重复的数使得其能够组成三角形的概率.

1 ≤ n , a i ≤ 1 e 5 1 \leq n,a_i \leq 1e5 1n,ai1e5

题目思路:

总方案数: C n 3 C^3_n Cn3

假设三个数 a ≤ b ≤ c a \leq b \leq c abc,那么显然可以构造多项式自乘得出 x k x^k xk的系数代表 a + b = k a+b=k a+b=k的方案数。

但是这个方案数有 a = b a=b a=b,先减去这个部分。
然后由于乘法有顺序,像 a b , b a ab,ba ab,ba这种,所以方案除以2,除掉顺序.

接下来枚举 c c c为最大值,考虑多项式的系数 > c > c >c的都可以。就是一个后缀和

但是这个时候发现还是有很多重复的部分要处理,比如你无法保证系数中的所有方案中的都 a , b a,b a,b都小于等于c.反向考虑不合法的情况:就是一个前缀和。这个时候就直接是方案了。即 a n s = ∑ p r e [ c i ] ans=\sum pre[c_i] ans=pre[ci]

PS:因为任意一个系数 ≤ c \leq c c,意味着 a + b ≤ c a+b \leq c a+bc.那么就可以保证 c c c是最大值,不会发生上面那种情况。

所以答案为 1 − a n s C n 3 1-\frac{ans}{C_n^3} 1Cn3ans.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define pb push_back
#define mp make_pair
#define vi vector<int>
#define vl vector<ll>
#define fi first
#define se second
namespace FFT{
    
    
    const int maxn = 302144+10;
    const double Pi = acos(-1.0);
    struct complex{
    
    
        double x,y;
        complex (double xx=0,double yy=0){
    
    x=xx,y=yy;}
    }a[maxn],b[maxn];
    complex operator + (complex a,complex b){
    
     return complex(a.x+b.x , a.y+b.y);}
    complex operator - (complex a,complex b){
    
     return complex(a.x-b.x , a.y-b.y);}
    complex operator * (complex a,complex b){
    
     return complex(a.x*b.x-a.y*b.y , a.x*b.y+a.y*b.x);}//不懂的看复数的运算那部分
    int l,r[maxn];
    int limit = 1;
    void fast_fast_tle(complex *A , int type)
    {
    
    
        for(int i=0;i<limit;i++)
            if(i<r[i]) swap(A[i],A[r[i]]);//求出要迭代的序列
        for(int mid=1;mid<limit;mid<<=1)//待合并区间的中点
        {
    
    
            complex Wn( cos(Pi/mid) , type*sin(Pi/mid) ); //单位根
            for(int R=mid<<1,j=0;j<limit;j+=R)//R是区间的右端点,j表示前已经到哪个位置了
            {
    
    
                complex w(1,0);//幂
                for(int k=0;k<mid;k++,w=w*Wn)//枚举左半部分
                {
    
    
                    complex x=A[j+k],y=w*A[j+mid+k];//蝴蝶效应
                    A[j+k]=x+y;
                    A[j+mid+k]=x-y;
                }
            }
        }
    }
    void ploy_mul (int x[] , int y[] , int n , int m , ll res[])
    {
    
    
        // 多组数据记得清空. 要清空到lim
        limit = 1;l = 0;
        while(limit<=n+m) limit<<=1 , l++;
        for (int i = 0 ; i <= limit ; i++)
            a[i].x = a[i].y = b[i].x = b[i].y = 0.0 , r[i] = 0;
        for (int i = 0 ; i < n ; i++) a[i].x = 1.0 * x[i] , a[i].y = 0.0;
        for (int i = 0 ; i < m ; i++) b[i].x = 1.0 * y[i] , b[i].y = 0.0;

        for(int i=0;i<limit;i++)
            r[i]= ( r[i>>1]>>1 )| ( (i&1)<<(l-1) ) ;
        fast_fast_tle(a,1);
        fast_fast_tle(b,1);
        for(int i=0;i<=limit;i++) a[i] = a[i] * b[i];
        fast_fast_tle(a,-1);
        for (int i = 0 ; i <= n + m ; i++) res[i] = (ll)(a[i].x/limit + 0.5);
    }
}
const int maxn = 1e5 + 5;
const int mod = 1e9 + 7;
int a[maxn];
ll res[maxn * 2];
ll calc (ll x){
    
    
    return x * (x - 1) * (x - 2) / 6;
}
int main()
{
    
    
    int t; scanf("%d" , &t);
    while (t--){
    
    
        int n ; scanf("%d" , &n);
        vi b;
        for (int i = 1 ; i <= n ; i++){
    
    
            int x ; scanf("%d" , &x);
            a[x]++;
            b.pb(x);
        }
        int mx = *max_element(b.begin(),b.end());
        FFT::ploy_mul(a ,  a , mx + 1 , mx + 1 , res);
        int up = mx * 2;
        for (auto g : b) res[g + g]--;
        for (int i = 1 ; i <= up ; i++) res[i] /= 2;
        for (int i = 1 ; i <= up ; i++) res[i] += res[i - 1];
        ll tot = 0;
        for (auto g : b) tot += res[g];
        double ans = (1.0 * tot) / (1.0 * calc(n));
        printf("%.7f\n" , 1.0 - ans);
        // 撤销贡献
        for (auto g : b) a[g]--;
        for (int i = 0 ; i <= up ; i++) res[i] = 0;
    }
    return 0;
}

二.SPOJ TSUM Triple Sums

题目大意:

给你 n n n个数字,然后从中任意挑选三个不重复的数字并求和,让你输出和的所以可能以及对应取到每个和的方案数。

题目思路:

这题稍微难点.

如果没有限制重复,直接 A 3 ( x ) A^3(x) A3(x)即可。但是限制三个都不重复,可以想到容斥。

三个都不重复=所有情况-3至少两个重复+2至少三个重复。

容斥系数: 3 = C n 3 3=C_n^3 3=Cn3.因为两个重复那么会有三种可能: y x x yxx yxx, x y x xyx xyx, x x y xxy xxy.

但是发现如果是三个都一样的情况,会被多减掉两次,所以得加回来两倍.

构造三个多项式,FFT后容斥计算.

猜你喜欢

转载自blog.csdn.net/qq_35577488/article/details/115165781