[BZOJ5333][SDOI2018]荣誉称号 树型DP

首先发现最终的$a_i$一定和$a_{\lfloor\frac{i}{2^{k+1}}\rfloor}$同余。
那么只要把第$k+1$层及其之后的代价统计进前$k$层即可，也就是预处理出$v_{i,j}$表示前$k$层中的$a_i$在模$m$调整至$j$的代价。
然后只要设$f_{i,j}$表示点$i$到第$k$层的路径上$a$的和模$m$为$j$的最小代价，简单树型DP即可。
复杂度$O(n+2^k\cdot m^2)$。
代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#define NN 10000010
#define N 4100
#define M 210
#define ll long long
using namespace std;
unsigned int SA, SB, SC;
int p, A, B;
int n,K,m,a[NN],b[NN],hd[NN];
ll d[N][M],v[N][M],f[N][M];
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 work()
{
    int D=min((1<<(K+1))-1,n);
    memset(d,0,sizeof(d));
    memset(f,0x3f,sizeof(f));   
    for(int i=1;i<=n;i++)
    {
        a[i]%=m;
        if(i<=D) hd[i]=i;
        else hd[i]=hd[i/(D+1)];     
        d[hd[i]][a[i]]+=b[i];
    }
    for(int i=1;i<=D;i++)
    {
        ll sum=d[i][0];v[i][0]=0;
        for(int j=1;j<m;j++)
            v[i][0]+=d[i][j]*(m-j),sum+=d[i][j];
        for(int j=1;j<m;j++)
            v[i][j]=v[i][j-1]+sum-d[i][j]*m;    
    }
    for(int i=D+1;i<(1<<(K+1));i++)
        for(int j=0;j<m;j++) f[i][j]=0;
    for(int i=(1<<(K+1));i<(D+1)<<1;i++)
        f[i][0]=0;  
    for(int i=D;i;i--)
        for(int j=0;j<m;j++)
            for(int k=0;k<m;k++)
                f[i][j]=min(f[i][j],f[i<<1][(j-k+m)%m]+f[i<<1|1][(j-k+m)%m]+v[i][k]);
    printf("%lld\n",f[1][0]);           
}
int main()
{
    int ca;
    scanf("%d",&ca);
    while(ca--)
    {
        gen();
        work();
    }
    return 0;
}