【bitset 技巧 分块】bzoj5087: polycomp

神仙zq发现了${n^2\sqrt n}\over 32$做法

Description

你有三个系数为0,1的多项式f(x),g(x),h(x)

求f(g(x)) mod h(x)

为方便起见，将答案多项式所有系数对2取模输出即可

如果f(x)=Sigma(Ak * X^k)

则f(g(x))=Sigma(Ak(g(x))^K

Input

一共三行，每行一个多项式,分别为f,g,h

对于一个多项式描述为n P0,P1...Pn其中Pi为0或1

多项式P(x)=P₀+P₁*x+....+P_n*xⁿ

记n表示多项式最高项的次数,n<=4000

Output

用同样的格式输出答案多项式

如果答案为0，输出0 0

---

题目分析

陈老师神题x1

观察到这里多项式的所有操作都是在系数$\mod 2$的意义下的，因此可以用bitset来加速多项式的一些操作。例如$O(n^2)$实现多项式取模。

void mod(poly &a, int pos)
{
    for (int i=pos; i>=p; i--)
        if (a[i]) a ^= c<<(i-p), a[i] = 0;　　//我第一次居然把标红地方给忘了
}

但是如同很多bitset的技巧题一样，非常重要的一点是bitset每次整体操作的复杂度是  $O(size)$  的。

这意味着$Poly\, +:O(n), \, Poly\, *:O(n^2)$

接下去我们从暴力开始谈起。

暴力做法　$O({{n^3}\over 32})$

第一个需要解决的问题是：$f(g(x))$。那么我们只需要对$g(x)$求$k$次幂（也即最暴力地k次自乘），再将这些结果相加得到多项式$f(g(x))$。至于取模的过程，则可以在每次multiply的时候顺带模干净，这样最终相加得到的结果就是在模多项式意义下的答案。

void mod(poly &a, int pos)　　　　//pos是a的度数
{
    for (int i=pos; i>=p; i--)
        if (a[i]) a ^= c<<(i-p), a[i] = 0;
}
void mult(poly a, poly b, poly &ret)　　　　//ret=a*b
{
    ret.reset();
    for (int i=0; i<=p; i++)
        if (a[i]) ret ^= b<<i;　　//这里就是模拟n^2多项式乘法的过程
    mod(ret, p<<1);
}

总的代码：

#include<bits/stdc++.h>
const int maxn = 8035;
typedef std::bitset<maxn> poly;

int n,m,p;
poly a,b,c,tmp,cnt;

void input(poly &a, int &n)
{
    scanf("%d",&n);
    for (int i=0, x; i<=n; i++)
    {
        scanf("%d",&x);
        if (x) a.set(i);
    }
}
void mod(poly &a, int pos)
{
    for (int i=pos; i>=p; i--)
        if (a[i]) a ^= c<<(i-p), a[i] = 0;
}
void mult(poly a, poly b, poly &ret)
{
    ret.reset();
    for (int i=0; i<=p; i++)
        if (a[i]) ret ^= b<<i;
    mod(ret, p<<1);
}
int main()
{
    input(a, n), input(b, m), input(c, p);
    tmp[0] = 1, mod(b, m);
    for (int i=0; i<=n; i++)
    {
        if (a[i]) cnt ^= tmp;
        mult(tmp, b, tmp);　　　　//复杂度n^3在这里
    }
    while (p>=0&&!cnt[p]) --p;
    if (p==-1) puts("0 0");
    else{
        printf("%d",p);
        for (int i=0; i<=p; i++)
            printf(" %d",cnt[i]?1:0);
    }
    return 0;
} 

对系数按10位分块　$O({{n^3}\over 320})$

参见法老博客：[BITSET 分块] BZOJ5087. polycomp

注：md[t]并不一定要等于0.这里的取模多项式最高位对计算无影响。

容易发现这种做法的复杂度的阶仍然是$n^3$.

对$i=a\sqrt k+b$分块　$O({{n^2\sqrt n}\over 32})$

233