首页 > 代码库 > bzoj1063【Noi2008】道路设计

bzoj1063【Noi2008】道路设计

题意:http://www.lydsy.com/JudgeOnline/problem.php?id=1063

   用一种划分方式将树划为重链和轻链,使得所有点到根节点的路径经过的轻链最大值最小

sol:  先判定图是否联通,若不连通输出-1

   考虑树形dp,f[i][j]表示以i为根的字树中到i的最大不便利值为j的方案数

   然而这时空都是O(n^2)的QAQ,而且没法转移

   考虑运用树链剖分的思想可证明,j之多为O(log2(n))的

   事实上,j在图为完全二叉树时取最大值O(log3(n))

   对于转移,f[i][j][k]表示以i为根值为j,i向儿子连了k条边的方案数

   技术分享

   则对于每个节点u,可以选择向(f1)或不向(f2)其某个儿子v连边

   则f1=f[v][j-1][0]+f[v][j-1][1]+f[v][j-1][2]  (不向儿子连边)

    f2=f[v][j][0]+f[v][j][1]  (向儿子连边)

   对于某个点,在遍历到他的一个儿子时,之前儿子的影响已经被累加进答案f[u][j][]

   考虑新加进来的点的影响(为避免后效性应按2-1-0的顺序计算):

    f[u][j][2]=f[u][j][2]*f1+f[u][j][1]*f2

    f[u][j][1]=f[u][j][1]*f1+f[u][j][0]*f2

    f[u][j][0]=f[u][j][0]*f1

   最后从小到大扫一遍j,如果有,输出即可,注意%Q等于0的情况(设为Q)

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
using namespace std;
const int Mx=200010;
int n,m,Q,tot,head[Mx],ver[Mx],next[Mx];
long long ans,f[Mx][12][3];
inline long long jud(long long tmp)//处理%Q=0的情况     
{
    if(tmp!=0&&tmp%Q==0) return Q;
    else return tmp%Q;
}
inline void add(int x,int y)
{
    tot++;
    next[tot]=head[x];
    ver[tot]=y;
    head[x]=tot;
}
void dfs(int x,int fa)
{
    int cnt=0;//cnt记录儿子个数
    for(int i=head[x];i;i=next[i])
        if(ver[i]!=fa)
            dfs(ver[i],x),cnt++;
    for(int j=0;j<=10;j++) f[x][j][0]=1;//f数组赋初值 
    if(cnt==0) return ;//叶子节点
    for(int i=head[x];i;i=next[i])//树形dp
        if(ver[i]!=fa)
        {
            int y=ver[i];
            for(int j=0;j<=10;j++)
            {
                long long f1,f2;
                f1=f[y][j-1][0]+f[y][j-1][1]+f[y][j-1][2];
                f2=f[y][j][0]+f[y][j][1];
                f[x][j][2]=jud((long long) f[x][j][2]*f1+(long long) f[x][j][1]*f2);
                f[x][j][1]=jud((long long) f[x][j][1]*f1+(long long) f[x][j][0]*f2);
                f[x][j][0]=jud((long long) f[x][j][0]*f1);
            }
        }
    
}
int main()
{
    scanf("%d%d%d",&n,&m,&Q);
    for(int i=1;i<=m;i++)
    {
        int x,y;scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    if(m<n-1) { puts("-1\n-1"); return 0; }
    dfs(1,0);
    for(int j=0;j<=10;j++)
    {
        ans=f[1][j][0]+f[1][j][1]+f[1][j][2];
        if(ans!=0)
        {
            cout<<j<<endl<<(ans%Q)<<endl;
            return 0;
        }
    }
    return 0;
}

bzoj1063【Noi2008】道路设计