多项式板子

不包含MTT的板子, 这种有的话就弃疗了

/*
求逆迭代 B = 2B - AB^2

系数反转 F(1 / x) * x ^ n

除法迭代 已知 F(x) = Q(x) * G(x) + R(x) F(x),G(x) 求Q(x) R(x)

F(x) = Q(x) * G(x) + R(x) mod x ^ {n + 1}

F(1 / x) = Q(1 / x) * G(1 / x) + R(1 / x) mod x ^ {n + 1}

Frev(x) = Qrev(x) * Grev(x) + Rrev(x) x ^(n - du_R) mod x ^ n + 1

deg_Q = n + m Deg_R < m
则 (n - dU_R > n - m)

F(x) = Q(x) * G(x) +  mod x ^ {n - m + 1}

然后求逆算出Q, 余数减去即可

多项式求导和积分 非常简单

多项式求ln 基本求导法则

G(x) = Ln(F(x))
G(x)‘ = Ln(F(x))‘ F(x)‘ = F(x)‘ / F(x)
没了

多项式exp

F(x) = e ^ (A(x))

ln(F(x)) - A(x) = 0

折G(x) = (ln(F(x)) - A(x))
则
F(x) = F0(x) - (G(F0(x)) / G‘(F0_(x))
= F0(x) -  (Ln(F0(x)) - A(x)) * F0(x)
= F0(x) (1 - ln(F0(x)) + A(x))
*/
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<queue>
#include<cmath>
#define ll long long
#define mmp make_pair
using namespace std;
int read() {
    int nm = 0, f = 1;
    char c = getchar();
    for(; !isdigit(c); c = getchar()) if(c == ‘-‘) f = -1;
    for(; isdigit(c); c = getchar()) nm = nm * 10 + c - ‘0‘;
    return nm * f;
}
const int M = (1 << 20) + 5;
const int  mod = 998244353;
int add(int a, int b) {
    a += b;
    a -= a >= mod ? mod : 0;
    a += a < 0  ? mod : 0;
    return a;
}

int mul(int a, int b) {
    return 1ll * a * b % mod;
}

int poww(int a, int b) {
    int ans = 1, tmp = a;
    for(; b; b >>= 1, tmp = mul(tmp, tmp)) if(b & 1) ans = mul(ans, tmp);
    return ans;
}

namespace Poly {
    const int g = 3;
    int n;
    int pw[M], rw[M];
    void getw() {
        for(int i = 1; i <= M; i <<= 1) {
            pw[i] = poww(g, (mod - 1) / 2 / i);
            rw[i] = poww(pw[i], mod - 2);
        }
    }

    void init(int m) {
        n = m;
        while(n > (n & -n)) n += (n & -n);
    }

    void fft(int *x, int dft) {
        for(int i = 0, j = 0; i < n; i++) {
            if(i < j) swap(x[i], x[j]);
            for(int l = n >> 1; (j ^= l) < l; l >>= 1);
        }
        for(int step = 1; step < n; step <<= 1) {
            int wn = (dft == 1) ? pw[step] : rw[step];
            for(int i = 0; i < n; i += step << 1) {
                int wnk = 1;
                for(int j = i; j < i + step; j++) {
                    int a = x[j], b = mul(wnk, x[j + step]);
                    x[j] = add(a, b);
                    x[j + step] = add(a, -b);
                    wnk = mul(wnk, wn);
                }
            }
        }
        if(dft == -1) {
            int invn = poww(n, mod - 2);
            for(int i = 0; i < n; i++) x[i] = mul(x[i], invn);
        }
    }

    void pcpy(int *x, int a, int *y, int b) {
        for(int i = 0; i < a; i++) {
            x[i] = (i < b) ? y[i] : 0;
        }
    }

    void pmul(int *x, int a, int *y, int b, int *z) {
        static int tx[M + 5], ty[M + 5];

        init(a + b - 1);
        pcpy(tx, n, x, a);
        pcpy(ty, n, y, b);
        fft(tx, 1), fft(ty, 1);
        for(int i = 0; i < n; i++) {
            tx[i] = mul(tx[i], ty[i]);
        }
        fft(tx, -1);
        for(int i = 0; i < a + b - 1; i++) {
            z[i] = tx[i];
        }
    }

    void pinv(int *x, int m, int *y) {
        static int tx[M + 5], ty[M + 5];
        if(m == 1) {
            y[0] = poww(x[0], mod - 2);
            return;
        }
        pinv(x, (m + 1) >> 1, y);

        init(m << 1);
        pcpy(tx, n, x, m);
        pcpy(ty, n, y, (m + 1) >> 1);

        fft(tx, 1), fft(ty, 1);
        for(int i = 0; i < n; i++) {
            tx[i] = 1ll * (2ll - 1ll * tx[i] * ty[i] % mod + mod) * ty[i] % mod;
        }

        fft(tx, -1);

        for(int i = 0; i < m; i++) {
            y[i] = tx[i];
        }
    }

    void prev(int *x, int m, int *y) {
        for(int i = 0; i < m; i++) y[i] = x[m - 1 - i];
    }

    void pdiv(int *x, int a, int *y, int b, int *z) {
        static int tx[M + 5], ty[M + 5], tz[M + 5];

        prev(x, a, tx);
        prev(y, b, ty);

        for(int i = b; i < a - b + 1; ++i) ty[i] = 0;

        pinv(ty, a - b + 1, tz);
        pmul(tx, a - b + 1, tz, a - b + 1, ty);
        prev(ty, a - b + 1, z);
    }

    void pmod(int *x, int a, int *y, int b, int *z) {
        static int tx[M + 5];

        if(a < b) {
            for(int i = 0; i < a; i++) {
                z[i] = x[i];
            }
            return;
        }

        pdiv(x, a, y, b, tx);
        pmul(tx, a - b + 1, y, b, tx);
        for(int i = 0; i < b; i++) {
            z[i] = (x[i] - tx[i] + mod) % mod;
        }
    }

    void der(int *x, int m, int *y) {
        for(int i = 1; i < m; i++) y[i - 1] = 1ll * i * x[i] % mod;
        y[n - 1] = 0;
    }

    void inte(int *x, int m, int *y) {
        for(int i = 1; i < m; i++) y[i] = 1ll * x[i - 1] * poww(i, mod - 2) % mod;
        y[0] = 0;
    }

    void ln(int *x, int m, int *y) {
        static int tx[M + 5], ty[M + 5], tz[M + 5];
        der(x, m, tx);
        pinv(x, m, ty);
        pmul(tx, m, ty, m, tz);
        inte(tz, m, y);
    }
// F0(x) (1 - ln(F0(x)) + A(x))

    void exp(int *x, int m, int *y) {
        static int ta[M + 5], tb[M + 5], tc[M + 5];
        if(m == 1) {
            y[0] = 1;
            return;
        }
        exp(x, (m + 1) >> 1, y);
        pcpy(ta, m, y, (m + 1) >> 1);
        ln(ta, m, tb);
        for(int i = 0; i < m; i++) {
            tb[i] = add(x[i], -tb[i]);
        }
        tb[0] = add(tb[0], 1);
        pcpy(ta, (m + 1) >> 1, y, (m + 1) >> 1);
        pmul(ta, (m + 1) >> 1, tb, m, tc);
        pcpy(y, m, tc, m);
    }
}

int pool[M * 20], *cur = pool;

struct poly {
    int *a, l;
    poly() {}
    poly(int x) {
        a = cur;
        cur += (l = x);
    }
    poly(int *b, int x) {
        a = cur, cur += (l = x);
        memcpy(a, b, sizeof (int) * l);
    }
};

poly operator * (const poly &a, const poly &b) {
    poly c(a.l + b.l - 1);
    Poly::pmul(a.a, a.l, b.a, b.l, c.a);
    return c;
}

poly operator % (const poly &a, const poly &b) {
    poly c(b.l - 1);
    Poly::pmod(a.a, a.l, b.a, b.l, c.a);
    return c;
}

poly operator / (const poly &a, const poly &b) {
    poly c(b.l - 1);
    Poly::pdiv(a.a, a.l, b.a, b.l, c.a);
    return c;
}

poly ln(poly x) {
    poly y(x.l);
    Poly::ln(x.a, x.l, y.a);
    return y;
}

poly exp_(poly x) {
    poly y(x.l);
    Poly::exp(x.a, x.l, y.a);
    return y;
}

int main() {
    Poly::getw();

    return 0;
}

原文地址:https://www.cnblogs.com/luoyibujue/p/10504996.html

时间: 2024-11-08 23:04:50

多项式板子的相关文章

多项式板子&#183;新

之前写过一个vector的多项式汇总...不过常数特别大 最近事儿比较少,准备写一个int[]的 目前没写完,先打了发多项式求逆,蛙了好几发... 最后发现递归下去回来之后忘了清系数了... int[]版的主要是数组大小和高位系数比较恶心,还有数组清空啥的... 不过常数肯定比vector小很多,所以还是尝试写一下好了 #include <cstdio> #include <algorithm> using namespace std; const int p = 99824435

多项式板子(待补充)

已经实现的操作 乘法NTT, mul 求逆inverse 积分integral,微分deriv 对数ln 指数exp 快速幂power(常数项为\(1\)) 开根sqrt(常数项为\(1\)) 说明 所有操作都是原地操作,以后有空改(大量copy还是比较麻烦.耗时) 常数还可以 Code const int N = 262144, G = 3, iG = 332748118, p = 998244353, maxlen = N; inline int add(int x, int y){retu

多项式快速插值

200+行的多项式板子题真爽啊 给定\(n\)个点的点值\((x_i,y_i)\),求这\(n\)个点确定的\(n-1\)次多项式 \(n\le 10^5\) 前置知识: 多项式多点求值 拉格朗日插值 微积分基础 首先我们有一个\(n^2\)的拉格朗日插值法 \[f(x)=\sum\limits_{i=1}^{n}y_i\prod\limits_{i\ne j}\frac{x-x_j}{x_i-x_j}\] 然后我们学习一个WC2017挑战就过了 考虑优化,我们知道这个形式它很死,把它变成重心插

多项式&amp;生成函数

多项式和生成函数 多项式 鸽掉了多项式开根(加强版). 这里放一份多项式板子合集 多项式乘法 背个板子就好了. FFT 泰勒展开与麦克劳林级数 若\(f(x)\)在\(x=x0\)处存在\(n\)阶导,那么: \[ \begin{align} f(x)&=f(x0)+\frac{f^1(x0)}{1!}(x-x0)+\frac{f^2(x0)}{2!}(x-x0)^2+...+\frac{f^n(x0)}{n!}(x-x0)^n+\xi\&=\sum_{i=0}^n\frac{f^i(x0

自选题口胡

上一个坑因为后来时间太紧,于是每天做做做,根本不想写题解了.这次我决定大概保持一天一题到两题的节奏,就不着急啦.(大概不会鸽啦 大概顺序是各处看看哪里有简单题,找找简单题做,无实力选手根本做不动题啊. 119:众数MAX 简单题,显然是卷积状物,不过贡献取决于较小的一项.按照套路根号分治下,大的之间的贡献暴力,小的部分枚举一个固定的值 x 从 1 到阈值,每次把所有 >=x 的拿出来卷积.发现卷积部分跑的很慢,最后大概优化到了 200ms 吧,阈值大概是 6,. 127:Ball cf 原题加强

【全家福】多项式的各种板子

写完帕秋莉的超级多项式于是正好贴个模板大汇总(带优化的那种...) //by Judge #include<bits/stdc++.h> #define Rg register #define fp(i,a,b) for(Rg int i=(a),I=(b)+1;i<I;++i) #define fd(i,a,b) for(Rg int i=(a),I=(b)-1;i>I;--i) #define ll long long using namespace std; const in

UOJ #34 多项式乘法

题目链接:多项式乘法 保存一发FFT与NTT板子. 学习链接:从多项式乘法到快速傅里叶变换 FFT NTT 注意差值回来的时候不取反也是可以的,只不过需要把数组\(reverse\)一下(根据单位复数根的性质应该不难理解) 代码(FFT): #include<iostream> #include<cstdio> #include<cstring> #include<algorithm> #include<cmath> #include<co

【uoj34】 多项式乘法

http://uoj.ac/problem/34 (题目链接) 题意 求两个多项式的乘积 Solution 挂个FFT板子.当然,是hzwer的板子→_→ 细节 FFT因为要满足n是2的幂,所以注意数组大小. 代码 // uoj34 #include<algorithm> #include<iostream> #include<cstdlib> #include<cstring> #include<complex> #include<cst

[学习笔记] 多项式与快速傅里叶变换(FFT)基础

引入 可能有不少OIer都知道FFT这个神奇的算法, 通过一系列玄学的变化就可以在 $O(nlog(n))$ 的总时间复杂度内计算出两个向量的卷积(或者多项式乘法/高精度乘法), 而代码量却非常小. 博主一年半前曾经因COGS的一道叫做"神秘的常数 $\pi$"的题目而去学习过FFT, 但是基本就是照着板子打打完并不知道自己在写些什么鬼畜的东西OwO 不过...博主这几天突然照着算法导论自己看了一遍发现自己似乎突然意识到了什么OwO然后就打了一道板子题还1A了OwO再加上午考试差点AK