LibreOJ #2566.「SDOI2018」荣誉称号 树形dp

题意

好长不想写。

分析

把$\lfloor\frac{x}{2}\rfloor$看成$x$的父亲，这样我们就得到了一棵二叉树，问题就变成了对于二叉树上每一条长度为k+1的链，都要满足其权值和模m等于0。
发现对于某一条链，若我们将其往上移一个单位，然后把重合部分去掉，可以得到每个点都和他的第k+1级祖先模m同余。
那么我们就可以把那些同余的点并起来，这样就只剩下那些深度不超过k的点需要处理。
预处理出$w_{x,i}$表示所有$x$的“附属节点”的权值都到达了$i$需要的最小代价，然后就可以进行dp，设$f_{x,i}$表示所有在$x$的子树中的叶节点到$x$形成的链权值均为$i$的最小代价，每次可以$O(m)$进行转移，显然$f_{1,0}$就是答案。
而预处理$w_{x,i}$的话，不难发现一个附属点对该点的$w$的贡献是一个等差数列，那么可以通过二次查分来实现快速预处理。
时间复杂度$O(n+2^km^2)$

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>

typedef long long LL;

const int N=10000005;
const LL inf=(LL)1e16;

int n,m,k,a[N],b[N],bel[N],dep[N],mx[N],last[N],cnt,top,stack[N];
LL w[4005][405],f[4005][205];
struct edge{int to,next;}e[N];

unsigned int SA, SB, SC;int p, A, B;
unsigned int rng61()
{
    SA ^= SA << 16;
    SA ^= SA >> 5;
    SA ^= SA << 1;
    unsigned int t = SA;
    SA = SB;
    SB = SC;
    SC ^= t ^ SA;
    return SC;
}
void gen()
{
    scanf("%d%d%d%d%u%u%u%d%d", &n, &k, &m, &p, &SA, &SB, &SC, &A, &B);
    for(int i = 1; i <= p; i++)scanf("%d%d", &a[i], &b[i]);
    for(int i = p + 1; i <= n; i++)
    {
        a[i] = rng61() % A + 1;
        b[i] = rng61() % B + 1;
    }
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void pre(int x,int fa)
{
    stack[++top]=x;dep[x]=dep[fa]+1;mx[x]=dep[x];
    if (top>k+1) bel[x]=bel[stack[top-k-1]];
    else
    {
        bel[x]=x;
        for (int i=0;i<=m*2;i++) w[x][i]=0;
    }
    for (int i=last[x];i;i=e[i].next)
        pre(e[i].to,x),mx[x]=std::max(mx[x],mx[e[i].to]);
    top--;
}

int get_dis(int x,int y)
{
    return x<y?y-x:m-x+y;
}

void dp(int x)
{
    int son1=0,son2=0;
    for (int i=last[x];i;i=e[i].next)
    {
        if (dep[e[i].to]>k+1||mx[e[i].to]<=k) continue;
        son2=son1;son1=e[i].to;
        dp(e[i].to);
    }
    if (dep[x]==k+1)
    {
        for (int i=0;i<m;i++) f[x][i]=w[x][i];
    }
    else
    {
        for (int i=0;i<m;i++)
        {
            f[x][i]=inf;
            for (int j=0;j<m;j++)
                f[x][i]=std::min(f[x][i],f[son1][j]+f[son2][j]+w[x][(i+m-j)%m]);
        }
    }
}

void clear()
{
    cnt=0;
    for (int i=1;i<=n;i++) last[i]=0;
}

int main()
{
    int T;scanf("%d",&T);
    while (T--)
    {
        clear();
        gen();
        for (int i=2;i<=n;i++) addedge(i>>1,i);
        pre(1,0);
        for (int i=n;i>=1;i--)
        {
            a[i]%=m;
            w[bel[i]][a[i]+1]+=b[i];
            w[bel[i]][a[i]+m]-=(LL)b[i]*m;
            w[bel[i]][a[i]+m+1]+=(LL)b[i]*(m-1);
            if (dep[i]>k+1||mx[i]<=k) continue;
            for (int j=1;j<=m*2;j++) w[i][j]+=w[i][j-1];
            for (int j=1;j<=m*2;j++) w[i][j]+=w[i][j-1];
            for (int j=0;j<m;j++) w[i][j]+=w[i][j+m];
        }
        dp(1);
        printf("%lld\n",f[1][0]);
    }
    return 0;
}