【51Nod1773】A国的贸易 解题报告

【51Nod1773】A国的贸易

Description

给出一个长度为 \(2^n\) 的序列，编号从\(0\)开始。每次操作后，如果 \(i\) 与 \(j\) 的二进制表示只差一位则第 \(i\) 个数会加上操作前的第 \(j\) 个数。求 \(t\) 次操作后序列中的每个数是多少。

Input

第一行两个正整数 \(n\) ,\(t\)，意义如题。
第二行 \(2^n\) 个非负整数，第 \(i\) 个数表示编号为 \(i-1\) 的城市的初始货物存储量。
\(n<=20 \ t<=10^9\)

Output

输出一行 \(2^n\) 个非负整数。
第 \(i\) 个数表示过了 \(t\) 天后，编号为 \(i-1\) 的城市上的货物数量对 \(1e9+7\) 取模的结果。

---

构造一个多项式\(F\)，\(F_i=1\)，当\(i\)是\(2\)的正整数幂的时候。

然后答案就是\(D*F^t\)，乘法是异或子集和，用\(FWT\)就可以了。

注意到不要写多项式快速幂，子集运算的长度是封闭的，所以化成类似点乘的形式后做快速幂即可，这样复杂度是\(O(n2^n)\)的

---

Code:

#include <cstdio>
const int N=(1<<20)+10;
const int mod=1e9+7,inv=500000004;
namespace Fast{
    const int LEN=10000000;
    char inp[LEN],outp[LEN];
    int tmp[20];
    int inpos,outpos;
    void init(){
        fread(inp,1,LEN,stdin);
        inpos=0; outpos=0;
    }
    char GetChar(){return inp[inpos++];}
    int read(){
        int ret=0; char ch=GetChar();
        while (ch<'0'||ch>'9') ch=GetChar();
        while ('0'<=ch&&ch<='9') ret=ret*10+ch-'0',ch=GetChar();
        return ret;
    }
    void PutChar(char ch){outp[outpos++]=ch;}
    void print(int x){
        int pos=0;
        if (!x) tmp[++pos]=0;
        else
            while (x) tmp[++pos]=x%10,x/=10;
        for (int i=pos;i>=1;--i) PutChar(tmp[i]+'0');
    }
    void Print(){fwrite(outp,1,outpos,stdout);}
}
#define mul(a,b) (1ll*(a)*(b)%mod)
#define add(a,b) ((a+b)%mod)
int n,len,t,T[N],D[N];
void xorfwt(int *a,int typ)
{
    for(int le=1;le<len;le<<=1)
        for(int p=0;p<len;p+=le<<1)
            for(int i=p;i<p+le;i++)
            {
                int tx=a[i],ty=a[i+le];
                a[i]=add(tx,ty),a[i+le]=add(tx,mod-ty);
                if(!typ) a[i]=mul(a[i],inv),a[i+le]=mul(a[i+le],inv);
            }
}
int qp(int d,int k){int f=1;while(k){if(k&1)f=mul(f,d);d=mul(d,d),k>>=1;}return f;}
int main()
{
    Fast::init();
    n=Fast::read(),t=Fast::read(),len=1<<n;
    for(int i=0;i<len;i++) T[i]=Fast::read();
    D[0]=1;for(int i=0;i<n;i++) D[1<<i]=1;
    xorfwt(T,1),xorfwt(D,1);
    for(int i=0;i<len;i++) T[i]=mul(T[i],qp(D[i],t));
    xorfwt(T,0);
    for(int i=0;i<len;i++) Fast::print(T[i]),Fast::PutChar(' ');
    Fast::Print();
    return 0;
}

---

2018.12.18