BZOJ1063树形dp

最新推荐文章于 2019-04-02 21:34:57 发布

sherlock_zhuang

最新推荐文章于 2019-04-02 21:34:57 发布

阅读量307

点赞数 1

分类专栏： my-code 文章标签：树形dp

本文链接：https://blog.csdn.net/sherlock_zhuang/article/details/50805886

版权

my-code 专栏收录该内容

28 篇文章 0 订阅

订阅专栏

思路:
网上题解蛮多的就不仔细讲了。
首先一个比较明显的想法是先确定最少要多少代价
再统计方案
然后首先贪心的想的话肯定是越深的节点越有可能会花费更多的代价
所以想办法让深层节点尽量在一条铁路上
于是用树链剖分可以想到最多有log(n)条状态数不多
所以用树形dp即可
f[i][j][k]分别表示节点,当前代价,与子树的连边情况(0,1,2)具体转移看code

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
using namespace std;
typedef long long LL;
const int imax=100000+229;
const int bmax=imax*2;
int n,m,Q,Max; 
int num,head[imax],to[bmax],inext[bmax];
LL f[imax][22][4];

void add(int u,int v){ to[num]=v; inext[num]=head[u]; head[u]=num++; }
void iread()
{
    scanf("%d%d%d",&n,&m,&Q);
    int a,b;
    memset(head,-1,sizeof(head));
    for(int i=1;i<=m;i++) 
    {   
        scanf("%d%d",&a,&b);
        add(a,b); add(b,a);
    }   
    Max=(log(n)/log(2))+1;
}

LL get_val(LL val)
{ 
    if((val%Q)==0) { if(val==0) return 0; else return Q;}
    else return val%Q;  
}
//if val==Q then do not %Q or maybe you will think that ans do not exist; 
void dfs(int x,int fa)
{
    int son=false;
    for(int i=head[x];i!=-1;i=inext[i]) 
        if(fa!=to[i]) dfs(to[i],x),son=1;

    for(int i=0;i<=Max;i++) f[x][i][0]=1;//边界 
    if(son==false) return;

    for(int i=head[x];i!=-1;i=inext[i])
        if(fa!=to[i])
        { 
            int v=to[i];
            for(int j=0;j<=Max;j++)
            { 
                LL f1=0,f2=0; //f1表示不向这棵子树连边,f2表示向这棵子树连边    
                if(j) f1=f[v][j-1][0]+f[v][j-1][1]+f[v][j-1][2];
                //如果不连边的话不管怎么都能继承 而且代价+1
                f2=f[v][j][0]+f[v][j][1]; //连边的话这个子节点不能已经连两条边,cost不变;
                /* 要注意顺序 这样写不行 
                f[x][j][0]=get_val(f[x][j][0]*f1); 
                f[x][j][1]=get_val(f[x][j][1]*f1+f[x][j][0]*f2);    
                f[x][j][2]=get_val(f[x][j][2]*f1+f[x][j][1]*f2);                
                */
                f[x][j][2]=get_val(f[x][j][2]*f1+f[x][j][1]*f2);
                f[x][j][1]=get_val(f[x][j][1]*f1+f[x][j][0]*f2);
                f[x][j][0]=get_val(f[x][j][0]*f1); 
            }
        }
    return;
}

int cnt,vis[imax];
queue<int> q;
bool bfs()
{
    vis[1]=1; q.push(1);
    while(!q.empty())
    {
        int u=q.front();
        q.pop(); cnt++;
        for(int i=head[u];i!=-1;i=inext[i])
            if(!vis[to[i]]) 
            {
                vis[to[i]]=1;
                q.push(to[i]);  
            }   
    }   
    return cnt>=n;
}

void iwork()
{
    if(!bfs()) { puts("-1\n-1"); return;}
    dfs(1,0);
    for(int i=0;i<=Max;i++)
    {
        int x=1; 
        LL now=f[x][i][0]+f[x][i][1]+f[x][i][2];
        if(now) { printf("%d\n%lld\n",i,now%Q); break; }    
    }   
}

int main()
{
    iread();
    iwork();
    return 0;
}