Wannafly Winter Camp Day5 Div1 E题 Fast Kronecker Transform 转化为NTT或FFT

最新推荐文章于 2021-03-21 15:29:34 发布
最新推荐文章于 2021-03-21 15:29:34 发布
阅读量101 收藏
点赞数
原文链接：http://www.cnblogs.com/Cwolf9/p/10322102.html
版权

目录

（有任何问题欢迎留言或私聊 && 欢迎交流讨论哦

Catalog

@

Problem：传送门

 原题目描述在最下面。
 对给定的式子算解。
 \(0\leq k\leq n+m,c_k=(\sum_{i+j=k}i\times j\times \sigma_{a_i,b_j}) mod\;998244353\)，其中\(当且仅当a=b时，\sigma_{a_i,b_j}=1。\)

Solution：

 我们发现只有当\(a_i\)和\(b_j\)相等时才会对答案造成贡献。

 一共\(2e^5\)个数字，我们枚举每一个数字算贡献，同时分情况讨论：当这个
数字出现次数小于阀值\(T\)时，我们\(O(n^2)\)暴力算；当出现次数大于\(T\)时，我们用\(FFTorNTT\)计算。

 听说这题有人\(FFT\)丢精度就\(wa\)了，我就干脆用\(NTT\)写了，刚好这个模数也是费马质数嘛，接下来看怎么把上述奇怪的卷积转化成一个可以用\(FFT\)和\(NTT\)计算的卷积。

 我们把\(a\)数组和\(b\)数组中数字\(x\)所有出现的位置提出来放到新数组里面去，比如\(x\)出现在\(a\)的\(1,3,4\)位置，也出现在\(b\)的\(2,4,5,6\)位置。

 当\(k\)等于\(5\)时，数字\(x\)产生的贡献\(1*4,2*3\)
 当\(k\)等于\(6\)时，数字\(x\)产生的贡献\(1*5\)

 我们知道一般的卷积式子是长这样的：\(C_k=\sum_{i+j=k}A_i\times B_j\)

 哇哦，感觉这个式子和那个贡献好相似啊。

 我们就构造这样两个数组\(A_{a_i}=a_i,B_{b_j}=b_j\)可以得到：
\(A[]=\{0,1,0,3,4\},B[]=\{0,0,2,0,4,5,6\}\)

 我们对这两个数组求他们多项式乘法的结果，然后将结果的贡献累计到答案数组中就行啦。

 然后当阀值\(T\)为\(10000\)时，复杂度就可以承受了。

AC_Code：

感谢日天的板子
这题复杂度的分析dls讲题时分析的很清楚啦，一位群友的笔记截图在下面

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
namespace lh {
#define o2(x) (x)*(x)
    using namespace std;
    typedef long long LL;
    typedef unsigned long long uLL;
    typedef pair<int, LL> pii;
}

using namespace lh;
const int MX = 2e5 + 5;
//const int P = (479 << 21) + 1;
const int P = 998244353;
const int MOD = 998244353;
const int G = 3;
const int NUM = 20;
struct my_NTT {
    LL wn[NUM];
    LL a[MX << 1], b[MX << 1];
    LL pow (LL a, LL x, LL mod) {
        LL ans = 1;
        a %= mod;
        while (x) {
            if (x & 1) ans = ans * a % mod;
            x >>= 1;
            a = a * a % mod;
        }
        return ans;
    }
    //在程序的开头就要放
    void init() {
        for (int i = 0; i < NUM; i++) {
            int t = 1 << i;
            wn[i] = pow (G, (P - 1) / t, P);
        }
    }
    void Rader (LL F[], int len) {
        int j = len >> 1;
        for (int i = 1; i < len - 1; i++) {
            if (i < j) swap (F[i], F[j]);
            int k = len >> 1;
            while (j >= k) j -= k, k >>= 1;
            if (j < k) j += k;
        }
    }
    void NTT (LL F[], int len, int t) {
        Rader (F, len);
        int id = 0;
        for (int h = 2; h <= len; h <<= 1) {
            id++;
            for (int j = 0; j < len; j += h) {
                LL E = 1;
                for (int k = j; k < j + h / 2; k++) {
                    LL u = F[k];
                    LL v = E * F[k + h / 2] % P;
                    F[k] = (u + v) % P;
                    F[k + h / 2] = (u - v + P) % P;
                    E = E * wn[id] % P;
                }
            }
        }
        if (t == -1) {
            for (int i = 1; i < len / 2; i++) swap (F[i], F[len - i]);
            LL inv = pow (len, P - 2, P);
            for (int i = 0; i < len; i++) F[i] = F[i] * inv % P;
        }
    }
    void Conv (LL a[], LL b[], int len) {
        NTT (a, len, 1);
        NTT (b, len, 1);
        for (int i = 0; i < len; i++) a[i] = a[i] * b[i] % P;
        NTT (a, len, -1);
    }
    void gao (LL A[], LL B[], int n, int m, LL ans[]) {//0~n-1
        int len = 1;
        while (len < n + m) len <<= 1;
        for (int i = 0; i < n; i++) a[i] = A[i];
        for (int i = 0; i < m; i++) b[i] = B[i];
        for (int i = n; i < len; i++) a[i] = 0;
        for (int i = m; i < len; i++) b[i] = 0;
        Conv (a, b, len);
        for (int i = 0; i < len; i++) ans[i] = (ans[i]+a[i])%MOD;
    }
}ntt;
const int MXN = 2e5 + 5;
int n, m;
int ar[MXN], br[MXN];
LL A[MXN], B[MXN];
std::vector<int> all[MXN], bll[MXN];
LL ans[MXN];
void solve1(int id) {
    for(int i = 0; i < all[id].size(); ++i) {
        for(int j = 0; j < bll[id].size(); ++j) {
            ans[all[id][i]+bll[id][j]] += (LL)all[id][i] * bll[id][j];
            ans[all[id][i]+bll[id][j]] %= MOD;
        }
    }
}
void solve2(int id) {
    for(int i = 0; i <= n+m; ++i) A[i] = B[i] = 0;
    for(int i = 0; i < all[id].size(); ++i) A[all[id][i]] = all[id][i];
    for(int i = 0; i < bll[id].size(); ++i) B[bll[id][i]] = bll[id][i];
    ntt.gao(A, B, all[id].back()+1, bll[id].back()+1, ans);
}
int main(int argc, char const *argv[]) {
    scanf("%d%d", &n, &m); ++n, ++m;
    ntt.init();
    std::vector<int> vs;
    for(int i = 0; i < n; ++i) scanf("%d", &ar[i]), vs.push_back(ar[i]);
    for(int i = 0; i < m; ++i) scanf("%d", &br[i]), vs.push_back(br[i]);
    sort(vs.begin(), vs.end());
    vs.erase(unique(vs.begin(), vs.end()), vs.end());
    for(int i = 0, tmp; i < n; ++i) {
        tmp = lower_bound(vs.begin(), vs.end(), ar[i]) - vs.begin();
        all[tmp].push_back(i);
    }
    for(int i = 0, tmp; i < m; ++i) {
        tmp = lower_bound(vs.begin(), vs.end(), br[i]) - vs.begin();
        bll[tmp].push_back(i);
    }
    for(int i = 0; i < vs.size(); ++i) {
        if(all[i].size() + bll[i].size() <= 10000) solve1(i);
        else solve2(i);
    }
    for(int i = 0; i <= n + m-2; ++i) printf(i!=n+m-2?"%lld ":"%lld\n", ans[i]);
    return 0;
}


Problem Description：

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述



果然，我的\(FFTwa了\)

#include<bits/stdc++.h>
#define fi first
#define se second
#define pb push_back
namespace lh {
#define o2(x) (x)*(x)
    using namespace std;
    typedef long long LL;
    typedef unsigned long long uLL;
    typedef pair<int, LL> pii;
}

using namespace lh;
const int MX = 2e5 + 5;
//const int P = (479 << 21) + 1;
const int P = 998244353;
const int MOD = 998244353;
const int G = 3;
const int NUM = 20;

struct my_FFT {
    const double pi = acos (-1.0);
    static const int MM = MX * 4;
    int len, res[MM], mx; //开大4倍
    struct cpx {
        double r, i;
        cpx (double r = 0, double i = 0) : r (r), i (i) {};
        cpx operator+ (const cpx &b) {return cpx (r + b.r, i + b.i);}
        cpx operator- (const cpx &b) {return cpx (r - b.r, i - b.i);}
        cpx operator* (const cpx &b) {return cpx (r*b.r - i*b.i,i*b.r + r * b.i);}
    } va[MM], vb[MM];
    void rader (cpx F[], int len) { //len = 2^M,reverse F[i] with  F[j] j为i二进制反转
        int j = len >> 1;
        for (int i = 1; i < len - 1; ++i) {
            if (i < j) swap (F[i], F[j]); // reverse
            int k = len >> 1;
            while (j >= k) j -= k, k >>= 1;
            if (j < k) j += k;
        }
    }
    void FFT (cpx F[], int len, int t) {
        rader (F, len);
        for (int h = 2; h <= len; h <<= 1) {
            cpx wn (cos (-t * 2 * pi / h), sin (-t * 2 * pi / h) );
            for (int j = 0; j < len; j += h) {
                cpx E (1, 0); //旋转因子
                for (int k = j; k < j + h / 2; ++k) {
                    cpx u = F[k];
                    cpx v = E * F[k + h / 2];
                    F[k] = u + v;
                    F[k + h / 2] = u - v;
                    E = E * wn;
                }
            }
        }
        if (t == -1) //IDFT
            for (int i = 0; i < len; ++i) F[i].r /= len;
    }
    void Conv (cpx a[], cpx b[], int len) { //求卷积
        FFT (a, len, 1);
        FFT (b, len, 1);
        for (int i = 0; i < len; ++i) a[i] = a[i] * b[i];
        FFT (a, len, -1);
    }
    void gao (LL a[], LL b[], int n, int m, LL ans[]) {
        len = 1;
        mx = n + m;
        while (len <= mx) len <<= 1; //mx为卷积后最大下标
        for (int i = 0; i < len; i++) va[i].r =va[i].i = vb[i].r = vb[i].i = 0;
        for (int i = 0; i < n; i++) va[i].r = a[i];//根据题目要求改写
        for (int i = 0; i < m; i++) vb[i].r = b[i];//根据题目要求改写
        Conv (va, vb, len);
        for (int i = 0; i < len; ++i) ans[i] = (ans[i]+(LL)(va[i].r + 0.5))%MOD;
    }

}fft;
const int MXN = 2e5 + 5;
int n, m;
int ar[MXN], br[MXN];
LL A[MXN], B[MXN];
std::vector<int> all[MXN], bll[MXN];
LL ans[MXN*10];
void solve1(int id) {
    for(int i = 0; i < all[id].size(); ++i) {
        for(int j = 0; j < bll[id].size(); ++j) {
            ans[all[id][i]+bll[id][j]] += (LL)all[id][i] * bll[id][j];
            ans[all[id][i]+bll[id][j]] %= MOD;
        }
    }
}
void solve2(int id) {
    for(int i = 0; i <= n+m; ++i) A[i] = B[i] = 0;
    for(int i = 0; i < all[id].size(); ++i) A[all[id][i]] = all[id][i];
    for(int i = 0; i < bll[id].size(); ++i) B[bll[id][i]] = bll[id][i];
    fft.gao(A, B, all[id].back()+1, bll[id].back()+1, ans);
}
int main(int argc, char const *argv[]) {
    scanf("%d%d", &n, &m); ++n, ++m;
    std::vector<int> vs;
    for(int i = 0; i < n; ++i) scanf("%d", &ar[i]), vs.push_back(ar[i]);
    for(int i = 0; i < m; ++i) scanf("%d", &br[i]), vs.push_back(br[i]);
    sort(vs.begin(), vs.end());
    vs.erase(unique(vs.begin(), vs.end()), vs.end());
    for(int i = 0, tmp; i < n; ++i) {
        tmp = lower_bound(vs.begin(), vs.end(), ar[i]) - vs.begin();
        all[tmp].push_back(i);
    }
    for(int i = 0, tmp; i < m; ++i) {
        tmp = lower_bound(vs.begin(), vs.end(), br[i]) - vs.begin();
        bll[tmp].push_back(i);
    }
    for(int i = 0; i < vs.size(); ++i) {
        if(all[i].size() + bll[i].size() <= 10000) solve1(i);
        else solve2(i);
    }
    for(int i = 0; i <= n + m-2; ++i) printf(i!=n+m-2?"%lld ":"%lld\n", ans[i]);
    return 0;
}

转载于:https://www.cnblogs.com/Cwolf9/p/10322102.html

ako2654
关注
【CCPC-Wannafly Winter Camp Day4 (Div2, onsite)】 G 置置置换
我是谁 --- 卡比兽
05-01 154
置置置换 题意 给你 n−1n - 1n−1长度的由&amp;gt;&amp;gt;>和&amp;lt;&amp;lt;<组成的序列 第偶数个括号是&amp;lt;&amp;lt;<代表 第i−1i-1i−1个数字比iii个数字大 奇数个和&amp;gt;&amp;gt;>同理 要满足 1 - n的全排列 所以我们设 dp[i][j] 代表 满足 1 - i 全排列 第 i...
2020 CCPC-Wannafly Winter Camp Day6 Div.1&2——C.酒馆战棋【模拟】
NIRVANA REBIRTH
02-02 381
题目传送门 题目描述 酒馆战棋是一个很有趣的游戏，这个游戏一共有两名玩家，每个玩家各自控制一定数量的随从，随从从左往右排列。 随从有两个基础属性：攻击力的血量，两个随从交战后会受到对方攻击力的伤害（失去对应的血量），当一个随从的血量降到 0{0}0 或者以下时会死亡。 整个对战流程如下：你的随从从左往右依次进行攻击，每次攻击是随机选择一个可以攻击的对方随从进行攻击。而对方的随从全程不动。 由于对...
CCPC-Wannafly Winter Camp Day5 E.Fast Kronecker Transform(离散化+NTT)
小衣同学的博客
01-25 259
题目 思路来源 dls 题解 先离散化，把a和b搞进一个序列，然后排序去重，把a和b赋为rank值 然后遍历这个不重复的序列，两个大于1W就建2e5数列用NTT搞，否则直接暴力 既然要在ai和bj值相同，下标i+j==k时求i*j的值， 那么就把位置i和j分别放入值的vector里，对相同值的vector进行NTT 注意多项式乘法一定会遍历所有的情形，所以就令f(x)的第i位...
Codeforces Round #136 (Div. 1)E(尺取法+树状数组)
shaohang_的博客
08-16 181
题目链接：https://codeforces.com/contest/220/problem/E 题意：只保留a数列中1..l和r..n的数构成b数列，然后b数列的逆序对数小于等于k.问这样的l,r的对数。 分析：树状数组+尺取 枚举l,每次找到最小的满足题意的r，对答案的贡献是n-r+1,然后用两个树状数组，分别维护增加或者减少一个数的时候，前半段和后半段对逆序数的影响。 Ac cod...
CCPC-Wannafly Winter Camp Day5（div1 + div2 部分题解）
Lee_w_j__的博客
01-25 897
啊，来了五天（今天应该是第六天了）camp了，终于可以愉快的补题了，由于前面欠下的题目好像有点多，所以只能从后往前将题目以及题解补上了（希望能在有生之年把能补的题目补完吧，QAQ）。 第五天是dls场，dls对待菜鸡还是非常友好的，div2的题目简直是快乐无比，div1的就不快乐了，但还是得把能补的题目补一补。   A.Cactus Draw div1版本：给你一棵仙人掌，要将所有节点放到...
【翻译】Fast Fourier Transform
03-15 2732
这是Stefan Wörner的一篇介绍快速傅里叶变换的论文，原文下载在这里有：http://download.csdn.net/detail/u013012544/7045283，是北京大学本科生算法设计与分析2014年春季课程的选读论文之一。 【这篇论文中的公式图表太多了...所以有很多部分都是截图...】 以下是它的全文翻译： 瑞士联邦理工学院 快速傅里叶
CCPC-Wannafly Winter Camp Day5 Div2 部分题解
ccsu_cat
01-24 430
本来这几天单挑ccpc camp div2很爽，直到今天遇到了计算几何.............. A.Cactus Draw 签到题 #include&amp;lt;bits/stdc++.h&amp;gt; #define ll long long using namespace std; const int maxn=1005; vector&amp;lt;int&amp;gt;G[maxn]; struct ...
CCPC-Wannafly Winter Camp Day5 (Div2, onsite) A Cactus Draw
CaprYang的博客
01-24 390
题解 建图后将度最大的节点作为根 行坐标为节点深度 使用l数组记录每行最后一个点的位置每次++ xy坐标限制在n范围内即使是一条链也可以满足 兄弟节点在同一行的坐标并依次往后排列 最后按照节点序号输出答案 AC代码 #include &amp;amp;amp;lt;stdio.h&amp;amp;amp;gt; #include &amp;amp;amp;lt;bits/stdc++.h&amp;amp;amp;gt; using namespace std; typedef long lo.
2020 CCPC Wannafly Winter Camp Day5 Div.1&2 E题 (暴力）
菱形继承
03-21 145
题目传送门 #include<bits/stdc++.h> using namespace std; #define ll long long typedef pair<int,int> pii; const int N=1e5+10; ll n,a[N],b[5],ans; int main(){ cin>>n; for(int i=1;i<=n;i++)cin>>a[i]; for(int i=1;i<=4;i++)cin>..
2020 CCPC Wannafly Winter Camp Day1 Div1&2（重现赛）B题解
lin_rany的博客
01-20 484
链接：https://ac.nowcoder.com/acm/contest/3979/B 来源：牛客网 B,签到题。按m->1倒序来依次还原字符串。 #include<bits/stdc++.h> //#pragma GCC optimize(2) #include<iostream> #include<stdio.h> #include<st...
CCPC-Wannafly Winter Camp Day2 (Div2, onsite) Erase Numbers II
c3po
01-30 552
题目描述   给定 nn 个正整数 a_1, a_2, \cdots, a_na1​,a2​,⋯,an​，它们组成了序列 AA。 你的任务是删除其中 (n - 2)(n−2) 数字，使得 AA 中剩余的数字连起来的所表示的数字最大，并给出这个值。 严格来讲，如果剩余数字在 AA 中的下标为 p_1, p_2, \cdots, p_mp1​,p2​,⋯,pm​ (1 \leq p_1 &amp;am...
7-6 2020 CCPC Wannafly Winter Camp Day1 F. 乘法（二分）
_Griefs
08-24 310
给出一个长度为 n 的数列 和一个长度为 m 的数列 ，可以构造得到一个 n×m 的矩阵 C，其中 C​i,j​​ =A​i​​ ×Bj​​ 。 给出整数 K，你需要求出 C 中第 K 大的数的值。 输入格式： 第一行输入三个整数 n,m,K(1≤n,m≤10​5​​ ,1≤K≤n×m)。 第二行输入 n 个空格隔开的整数 A​1​​ ,…,A​n​​ (−10​6​​ ≤A​i​​ ≤10​6​...
2020 CCPC-Wannafly Winter Camp Day1_A 期望逆序对
Vibrant
01-22 520
贪心+数学
电动车上牌管理系统 SSM毕业设计 附带论文.zip
11-04
电动车上牌管理系统 SSM毕业设计 附带论文 启动教程:https://www.bilibili.com/video/BV1GK1iYyE2B
tornado-6.1-cp39-cp39-manylinux2010_x86_64.whl
11-04
tornado-6.1-cp39-cp39-manylinux2010_x86_64.whl
【eclipse和idea两个版本运行源码】基于Java Swing +mysql 实现的网吧管理系统
最新发布
11-04
一、项目简介 本项目是一套基于Java Swing 开发的网吧管理系统，主要针对计算机相关专业的正在做毕设的学生与需要项目实战练习的Java学习者。 包含：项目源码、数据库脚本等，该项目附带全部源码可作为毕设使用。 项目都经过严格调试，确保可以运行！ 该系统功能完善、界面美观、操作简单、功能齐全、管理便捷，具有很高的实际应用价值。 二、技术实现 ​后台技术：java swing ​数据库：MySQL ​数据库连接池：c3p0 三、系统主要功能 用户登录: 分为 普通用户和管理员 两种角色 菜单模块：上机，下机， 系统设置：管理员设置，会员设置，计费设置， 退出系统 管理模块：增加会员，删除会员，信息修改，信息查询 视图模块：主页视图，在线用户，统计视图， 统计报表模块：人数报表，收入报表 帮助模块：联系我们，关于系统 详见：https://blog.csdn.net/weixin_43860634/article/details/125247764
pc-dmis软件脚本-输出Excel格式报告
11-04
使用软件自带的basic脚本编辑制作的脚本 低版本软件无法输出Excel报告，可以通过脚本方式实现这一功能
【java毕业设计】校园失物招领系统源码（springboot+vue+mysql+说明文档）.zip
11-04
项目经过测试均可完美运行！ 环境说明： 开发语言：java jdk：jdk1.8 数据库：mysql 5.7+ 数据库工具：Navicat11+ 管理工具：maven 开发工具：idea/eclipse
基于java的网上电子书店h答辩PPT.pptx
11-04
基于java的网上电子书店h答辩PPT.pptx
基于微信小程序的微信小程序校园失物招领答辩PPT.pptx
11-04
基于微信小程序的微信小程序校园失物招领答辩PPT.pptx
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值