bzoj2286: [Sdoi2011消耗战] 虚树构造

首先我们考虑每次都做一遍树形DP(树形DP自己脑补去,随便乱搞就过了)。

显然这是TLE无疑的。

由于要改变的节点个数很少,我们可以考虑用虚树重新建图,缩小范围,这样一棵树上最多有m*2个节点

然后每次对虚树进行树形dp这样总的复杂度就是n*logn的了,logn是lca的复杂度。

然后就是各种细节的地方了。。。。。

#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long sint;
#define pii pair
#define mp make_pair
#define deg 20
#define maxn 501000
#define inf 0x3f3f3f3f
#define INF (1ll<<40)
int getint()
{
    int res;char c;
    while(c=getchar(),c<'0'||c>'9');
    res=c-'0';
    while(c=getchar(),c>='0'&&c<='9')
        res=res*10+c-'0';
    return res;
}
struct node
{
    int u,v,next;
}tree[maxn];
int first[maxn],next[maxn],to[maxn],fa[maxn][deg];
sint dp[maxn];
int indexs,en,en2,pre[maxn],n,dfn[maxn],dep[maxn],num,T;
pii g[maxn];
int vis[maxn],meet[maxn],val[maxn],minn[maxn][deg];
void build(int a,int b,int c)
{
    en++;
    to[en]=b;
    val[en]=c;
    next[en]=first[a];
    first[a]=en;
}
void add(int u,int v)
{
    en2++;
    tree[en2].v=v;
    if(vis[u]!=T)
    {
        vis[u]=T;
        tree[en2].next=0;
    }
    else
    {
        tree[en2].next=pre[u];
    }
    pre[u]=en2;
}
void dfs(int now)
{
    int v;
    indexs++;
    dfn[now]=indexs;
    for(int i=first[now];i;i=next[i])
    {
        v=to[i];
        if(v==fa[now][0]) continue;
        fa[v][0]=now;
        minn[v][0]=val[i];
        for(int j=1;j=0;i--)
    {
        if(dep[fa[u][i]]>=dep[v])
            u=fa[u][i];
    }
    if(u==v)return v;
    for(int i=19;i>=0;i--)
    if(fa[u][i]!=fa[v][i])
    {
        u=fa[u][i];
        v=fa[v][i];
    }
    return fa[u][0];
}
int sta[maxn],top;
inline long long getmin(int x,int y)
{
    int ans=inf;
    if(dep[x]=0;i--)
        if(dep[fa[x][i]]>=dep[y])
        {
            ans=min(ans,minn[x][i]);
            x=fa[x][i];
        }
    if(x==y)return ans;
    for(int i=19;i>=0;i--)
        if(fa[x][i]!=fa[y][i])
        {
            ans=min(ans,min(minn[x][i],minn[y][i]));
            x=fa[x][i],y=fa[y][i];
        }
    if(x!=y)ans=min(ans,min(minn[x][0],minn[y][0]));
    return ans;
}
void dfs2(int x)
{
    int v;
    dp[x]=0;
    if(vis[x]==T)
    for(int i=pre[x];i;i=tree[i].next)
    {
        dfs2(v=tree[i].v);
        dp[x]+=min(getmin(v,x),meet[v]==T?inf:dp[v]);
    }
}
void virtree()
{
    sta[top=1]=1;
    for(int i=1;i<=num;i++)
    {
        int j=g[i].second;
        int anc=lca(sta[top],j);
        for(;dep[anc]

你可能感兴趣的:(数据结构)