zoukankan      html  css  js  c++  java
  • BZOJ 3771 生成函数,FFT

    Description

    我们讲一个悲伤的故事。
    从前有一个贫穷的樵夫在河边砍柴。
    这时候河里出现了一个水神,夺过了他的斧头,说:
    “这把斧头,是不是你的?”
    樵夫一看:“是啊是啊!”
    水神把斧头扔在一边,又拿起一个东西问:
    “这把斧头,是不是你的?”
    樵夫看不清楚,但又怕真的是自己的斧头,只好又答:“是啊是啊!”
    水神又把手上的东西扔在一边,拿起第三个东西问:
    “这把斧头,是不是你的?”
    樵夫还是看不清楚,但是他觉得再这样下去他就没法砍柴了。
    于是他又一次答:“是啊是啊!真的是!”
    水神看着他,哈哈大笑道:
    “你看看你现在的样子,真是丑陋!”
    之后就消失了。
    樵夫觉得很坑爹,他今天不仅没有砍到柴,还丢了一把斧头给那个水神。
    于是他准备回家换一把斧头。
    回家之后他才发现真正坑爹的事情才刚开始。
    水神拿着的的确是他的斧头。
    但是不一定是他拿出去的那把,还有可能是水神不知道怎么偷偷从他家里拿走的。
    换句话说,水神可能拿走了他的一把,两把或者三把斧头。
    樵夫觉得今天真是倒霉透了,但不管怎么样日子还得过。
    他想统计他的损失。
    樵夫的每一把斧头都有一个价值,不同斧头的价值不同。总损失就是丢掉的斧头价值和。
    他想对于每个可能的总损失,计算有几种可能的方案。
    注意:如果水神拿走了两把斧头a和b,(a,b)和(b,a)视为一种方案。拿走三把斧头时,(a,b,c),(b,c,a),(c,a,b),(c,b,a),(b,a,c),(a,c,b)视为一种方案。

    Input

    第一行是整数N,表示有N把斧头。
    接下来n行升序输入N个数字Ai,表示每把斧头的价值。

    Output

    若干行,按升序对于所有可能的总损失输出一行x y,x为损失值,y为方案数。

    Sample Input

    4
    4
    5
    6
    7

    Sample Output

    4 1
    5 1
    6 1
    7 1
    9 1
    10 1
    11 2
    12 1
    13 1
    15 1
    16 1
    17 1
    18 1
    样例解释
    11有两种方案是4+7和5+6,其他损失值都有唯一方案,例如4=4,5=5,10=4+6,18=5+6+7.

    HINT

    所有数据满足:Ai<=40000

    题意:

    给出 n个物品,价值为别为Xi且各不相同,现在可以取1个、2个或3个,问每种价值和有几种情况?顺序不同算一种。

    解法:

    显然是个母函数,A表示每种物品取一个的情况,B表示每种物品取二个的情况,C表示每种物品取三个的情况。用指数表示价值,系数表示该价值的个数,显然多项式相乘后指数会相加,系数会相乘,很容易就求出来了。

    所以对于每种物品价值x,A[x]++,B[2*x]++,C[3*x]++。

    如果取1个物品,答案就是A。

    如果取2个物品,A^2中有重复的(x,x)的情况,所以答案为A^2-B。

    如果去3个物品,A^3中可能有(x,x,x)(x,x,y)(x,y,x)(y,x,x)这几种重复的情况,而A*B能够求出所有形容(x,x,x)和(x,y,y)的情况数。(x,x,y)(x,y,x)(y,x,x)总的情况数=(x,y,y)*3,而A*B*3又会多减去了两次(x,x,x),所以要用C加回来。所以答案为A^3-3*B*A+2C。又由于顺序不同算一种情况,因为每种物品价值都不一样,情况(2)/2,情况(3)/6。

    故总情况数量等于:

    #include <bits/stdc++.h>
    using namespace std;
    const int maxn = 300000;
    typedef long long LL;
    const double PI = acos(-1.0);
    typedef complex <double> Complex;
    
    void rader(Complex *y, int len) {
        for(int i = 1, j = len / 2; i < len - 1; i++) {
            if(i < j) swap(y[i], y[j]);
            int k = len / 2;
            while(j >= k) {j -= k; k /= 2;}
            if(j < k) j += k;
        }
    }
    void fft(Complex *y, int len, int op) {
        rader(y, len);
        for(int h = 2; h <= len; h <<= 1) {
            double ang = op * 2 * PI / h;
            Complex wn(cos(ang), sin(ang));
            for(int j = 0; j < len; j += h) {
                Complex w(1, 0);
                for(int 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(op == -1) for(int i = 0; i < len; i++) y[i] /= len;
    }
    
    Complex a[maxn],b[maxn],c[maxn];
    int n, len, x, m, mx;
    
    int main()
    {
        scanf("%d", &n);
        for(int i=0; i<n; i++){
            scanf("%d", &x);
            a[x]+=(1),b[2*x]+=(1),c[3*x]+=(1);
            mx = max(mx, 3*x);
        }
        mx++;
        len = 1;
        while(len < mx*2){
            len <<= 1;
        }
        m = len+1;
        fft(a, len, 1);
        fft(b, len, 1);
        fft(c, len, 1);
        Complex t2=(2),t3=(3),t6=(6);
        for (int i=0;i<len;i++)
            a[i]=(a[i]*a[i]*a[i]-t3*a[i]*b[i]+t2*c[i])/t6+(a[i]*a[i]-b[i])/t2+a[i];
        fft(a, len, -1);
        for(int i=1; i<m; i++){
            LL num = (LL)(a[i].real()+0.5);
            if(num!=0) printf("%d %lld
    ", i,num);
        }
        return 0;
    }
    
  • 相关阅读:
    字符串实现变量映射
    纯css实现无限嵌套菜单
    flex布局嵌套之高度自适应
    easy ui 零散技巧
    js高级应用
    前端利器
    css高级应用及问题记录(持续更新)
    兼容性验证方案
    原生js判断元素是否可见
    vue-cli webpack躺坑之旅
  • 原文地址:https://www.cnblogs.com/spfa/p/7410183.html
Copyright © 2011-2022 走看看