[BZOJ2286][SDOI2011]消耗战

7 篇文章 0 订阅
3 篇文章 0 订阅

dp[i] 表示 i 所在的子树里面所有的询问点与i不连通的最小代价,那么 dp[i]=min(dp[j],w[i][j]),json[i] ,直接dp会TLE,发现询问的点数很少,那么建立虚树就可以了,一般虚树的题都很明显的。。
虚树就是关键点以及它们两两LCA所形成的树。
我写的常数好大QAQ。

#include<ctime>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<cassert>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<sstream>
#include<climits>
#define X first
#define Y second
#define DB double
#define lc now<<1
#define rc now<<1|1
#define MP make_pair
#define LL long long
#define pb push_back
#define sqr(_) ((_)*(_))
#define int LL 
#define INF 0x3f3f3f3f
#define pii pair<int,int>
#define pdd pair<DB,DB>
#define ull unsigned LL
#define DEBUG(...) fprintf(stderr,__VA_ARGS__)
using namespace std;
template<typename T>void Read(T& x)
{
    x=0;int flag=0,sgn=1;char c;
    while(c=getchar())
    {
        if(c=='-')sgn=-1;
        else if(c>='0'&&c<='9')x*=10,x+=c-'0',flag=1;
        else if(flag)break;
    }
    x*=sgn;
}
const int MAXN=1000010;
int n,m,dep[MAXN],dfn[MAXN],fa[MAXN][21],dfs_clock,g[MAXN][21];
int k,h[MAXN],st[MAXN],top=0,dp[MAXN],pa[MAXN],iskey[MAXN];
vector<int> A;
struct Tree{
    int first[MAXN],next[MAXN],e,w[MAXN],to[MAXN];
    Tree(){
        memset(first,-1,sizeof(first));
        e=0;
    }
    void add(int u,int v,int c)
    {
        next[e]=first[u];first[u]=e;to[e]=v;w[e]=c;++e;
        next[e]=first[v];first[v]=e;to[e]=u;w[e]=c;++e;
    }
    void init()
    {
        e=0;
    } 
    void DFS(int u,int f,int c)
    {
        if(u==1)
            dep[u]=1;
        else dep[u]=dep[f]+1;
        dfn[u]=++dfs_clock;
        fa[u][0]=f;
        g[u][0]=c;
        for(int i=1;i<21;i++)
        {
            fa[u][i]=fa[fa[u][i-1]][i-1];
            g[u][i]=min(g[fa[u][i-1]][i-1],g[u][i-1]);
        }
        for(int i=first[u];i!=-1;i=next[i])
        {
            int v=to[i];
            if(v==f)continue;
            DFS(v,u,w[i]);
        }
    }
    int LCA(int a,int b)
    {
        if(dep[a]<dep[b])swap(a,b);
        for(int i=20;i>=0;i--)
            if(dep[fa[a][i]]>=dep[b])
                a=fa[a][i];
        if(a==b)return a;
        for(int i=20;i>=0;i--)
        {
            if(fa[a][i]&&fa[b][i]&&fa[a][i]!=fa[b][i])
                a=fa[a][i],
                b=fa[b][i];
        }
        return fa[a][0];
    }
    int getMin(int a,int b)
    {
        int res=INF;
        for(int i=20;i>=0;i--)
        {
            if(fa[a][i]&&dep[fa[a][i]]>=dep[b])
            {
                res=min(res,g[a][i]);
                a=fa[a][i];
            }
        }
        return res;
    }
}ori,vit;
bool cmp(const int& a,const int& b)
{
    return dfn[a]<dfn[b];
}
void DP(int u,int f)
{
    int tmp=0;
    if(iskey[u])
        return;
    for(int i=vit.first[u];i!=-1;i=vit.next[i])
    {
        if(vit.to[i]==f)continue;
        DP(vit.to[i],u);
        tmp+=min(dp[vit.to[i]],vit.w[i]);
    }
    dp[u]=tmp;
}
void build()
{
    top=0;
    vit.init();
    sort(h+1,h+k+1,cmp);
    st[++top]=1;
    for(int i=1;i<=k;i++)
    {
        iskey[h[i]]=1;
        if(!top)
        {
            pa[h[i]]=1;
            st[++top]=h[i];
            continue;
        }
        int x=ori.LCA(h[i],st[top]);
        A.pb(x);
        A.pb(h[i]);
        for(;dep[st[top]]>dep[x];top--)
            if(dep[st[top-1]]<=dep[x])
                pa[st[top]]=x;
        if(st[top]!=x)
        {
            pa[x]=st[top];
            st[++top]=x;
        }
        pa[h[i]]=x;
        st[++top]=h[i];
    }
    sort(A.begin(),A.end());
    A.erase(unique(A.begin(),A.end()),A.end());
    for(int i=0;i<A.size();i++)
        if(pa[A[i]])
        {
            vit.add(A[i],pa[A[i]],ori.getMin(A[i],pa[A[i]]));
            //DEBUG("%d %d %d\n",A[i],pa[A[i]],ori.getMin(A[i],pa[A[i]])); 
        }
    DP(1,0);
    printf("%lld\n",dp[1]);
    for(int i=0;i<A.size();i++)
    {
        vit.first[A[i]]=-1,pa[A[i]]=0;
        dp[A[i]]=INF;
        if(iskey[A[i]])
            iskey[A[i]]=0;
    }
    A.clear();
}
main()
{
#ifndef ONLINE_JUDGE
    freopen("consume.in","r",stdin);
    freopen("consume.out","w",stdout);
#endif
    memset(g,INF,sizeof(g));
    memset(dp,INF,sizeof(dp));
    Read(n);
    for(int i=1;i<n;i++)
    {
        int a,b,c;
        Read(a);Read(b);Read(c);
        ori.add(a,b,c);
    }
    ori.DFS(1,0,0);
    Read(m);
    for(int i=1;i<=m;i++)
    {
        Read(k);
        for(int j=1;j<=k;j++)
            Read(h[j]);
        build();
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值