【bzoj3611】 大工程

http://www.lydsy.com/JudgeOnline/problem.php?id=3611 (题目链接)

搞了1天= =,感觉人都变蠢了。。。

题意:给出一个n个节点的树,每条边边权为1,给出q个询问,每次询问K个关键点,求出这k个点之间的两两距离和、最小距离和最大距离。

solution 
  构造虚树,见 http://blog.csdn.net/MashiroSky/article/details/51971718 
  之后在虚树上dp,有点麻烦。 
  用size[u]表示在以u为根的子树上的关键点个数,b[u]表示虚树上的节点u是否是关键点,f[u]表示以u为根的子树上所有关键点到u的距离之和,mn[u]表示以u为根的子树上距离u最近的关键点到u的距离,mx[u]表示以u为根的子树上距离u最远的关键点到u的距离。

代码:

// bzoj3611
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<set>
#define MOD 1000000007
#define inf 2147483640
#define LL long long
#define free(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout);
using namespace std;
inline LL getint() {
    LL x=0,f=1;char ch=getchar();
    while (ch>‘9‘ || ch<‘0‘) {if (ch==‘-‘) f=-1;ch=getchar();}
    while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}

const int maxn=1000010;
struct edge {int next,to,w;}e[maxn<<2];
int deep[maxn],fa[maxn][20],bin[20],dfn[maxn],head[maxn];
int mn[maxn],mx[maxn],s[maxn],a[maxn],b[maxn];
int n,K,cnt,top,q;
LL tot,ans1,ans2,f[maxn],size[maxn];

void insert(int u,int v) {
    if (u==v) return;
    e[++cnt].to=v;e[cnt].next=head[u];head[u]=cnt;
    e[cnt].w=deep[v]-deep[u];
}
void dfs(int u) {
    dfn[u]=++cnt;
    for (int i=1;i<20;i++) fa[u][i]=fa[fa[u][i-1]][i-1];
    for (int i=head[u];i;i=e[i].next) if (e[i].to!=fa[u][0]) {
            deep[e[i].to]=deep[u]+1;
            fa[e[i].to][0]=u;
            dfs(e[i].to);
        }
    head[u]=0;
}
int lca(int x,int y) {
    if (deep[x]<deep[y]) swap(x,y);
    int t=deep[x]-deep[y];
    for (int i=0;bin[i]<=t;i++) if (t&bin[i]) x=fa[x][i];
    for (int i=19;i>=0;i--)
        if (fa[x][i]!=fa[y][i]) x=fa[x][i],y=fa[y][i];
    return x==y?x:fa[x][0];
}
bool cmp(int x,int y) {
    return dfn[x]<dfn[y];
}
void dp(int u) {
    size[u]=b[u];
    f[u]=0;
    mn[u]=b[u]?0:inf;
    mx[u]=b[u]?0:-inf;
    for (int i=head[u];i;i=e[i].next) {
        int v=e[i].to;
        dp(v);
        tot+=(f[u]+size[u]*e[i].w)*size[v]+f[v]*size[u];
        size[u]+=size[v];
        f[u]+=f[v]+e[i].w*size[v];
        ans1=min(ans1,(LL)mn[u]+mn[v]+e[i].w);
        ans2=max(ans2,(LL)mx[u]+mx[v]+e[i].w);
        mn[u]=min(mn[u],mn[v]+e[i].w);
        mx[u]=max(mx[u],mx[v]+e[i].w);
    }
    head[u]=0;
}
void build() {
    scanf("%d",&K);
    for (int i=1;i<=K;i++) scanf("%d",&a[i]),b[a[i]]=1;
    sort(a+1,a+1+K,cmp);
    s[++top]=1;
    for (int i=1;i<=K;i++) {
        int t=a[i],f=0;
        while (top>0) {
            f=lca(a[i],s[top]);
            if (top>1 && deep[f]<deep[s[top-1]])
                insert(s[top-1],s[top]),top--;
            else if (deep[f]<deep[s[top]]) {insert(f,s[top--]);break;}
            else break;
        }
        if (s[top]!=f) s[++top]=f;
        s[++top]=t;
    }
    while (--top) insert(s[top],s[top+1]);
}
int main() {
    bin[0]=1;for (int i=1;i<20;i++) bin[i]=bin[i-1]<<1;
    scanf("%d",&n);
    for (int i=1;i<n;i++) {
        int u=getint(),v=getint();
        insert(u,v);insert(v,u);
    }
    cnt=0;
    dfs(1);
    scanf("%d",&q);
    while (q--) {
        build();
        ans1=inf;ans2=-inf;tot=0;
        dp(1);
        printf("%lld %lld %lld\n",tot,ans1,ans2);
        for (int i=1;i<=K;i++) b[a[i]]=0;
    }
    return 0;
}

  

时间: 2024-10-09 00:12:45

【bzoj3611】 大工程的相关文章

【BZOJ3611】大工程(虚树,动态规划)

[BZOJ3611]大工程(虚树,动态规划) 题面 BZOJ Description 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,我们选中了 k 个点,然后在它们两两之间 新建 C(k,2)条 新通道. 现在对于每个计划,我们想知道: 1.这些新通道的代价和 2.这些新通道中代价最小的是多

刷题总结——大工程(bzoj3611)

题目: Description 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,我们选中了 k 个点,然后在它们两两之间 新建 C(k,2)条 新通道. 现在对于每个计划,我们想知道: 1.这些新通道的代价和 2.这些新通道中代价最小的是多少 3.这些新通道中代价最大的是多少 Input 第一

[Bzoj3611][Heoi2014]大工程(虚树)

3611: [Heoi2014]大工程 Time Limit: 60 Sec  Memory Limit: 512 MBSubmit: 2000  Solved: 837[Submit][Status][Discuss] Description 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,

BZOJ3611 [Heoi2014]大工程 【虚树】

题目 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,我们选中了 k 个点,然后在它们两两之间 新建 C(k,2)条 新通道. 现在对于每个计划,我们想知道: 1.这些新通道的代价和 2.这些新通道中代价最小的是多少 3.这些新通道中代价最大的是多少 输入格式 第一行 n 表示点数. 接下来

bzoj 3611: [Heoi2014]大工程

Description 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,我们选中了 k 个点,然后在它们两两之间 新建 C(k,2)条 新通道. 现在对于每个计划,我们想知道: 1.这些新通道的代价和 2.这些新通道中代价最小的是多少 3.这些新通道中代价最大的是多少 Input 第一行 n

大工程(bzoj 3611)

Description 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,我们选中了 k 个点,然后在它们两两之间 新建 C(k,2)条 新通道. 现在对于每个计划,我们想知道: 1.这些新通道的代价和 2.这些新通道中代价最小的是多少 3.这些新通道中代价最大的是多少 Input 第一行 n

LibreOJ #2219. 「HEOI2014」大工程

二次联通门 : LibreOJ #2219. 「HEOI2014」大工程 /* LibreOJ #2219. 「HEOI2014」大工程 虚树 + dp 对于每次的关键点建好虚树后 考虑树形dp dp[i] 表示在以i为根的子树路径总长 size[i] 表示在以i为根的子树中关键点的个数 maxn为以i为根的子树中到关键点到根最长的路径长 ans1自己推一推就好 对于ans2 如果i是关键点,则直接用maxn[i]更新答案 否则用maxn[i]+maxn[child[i]]+dis(i,son[

Android Studio编译大工程报错:java.exe&#39;&#39; finished with non-zero exit value 1

版权声明:本文为博主原创文章,未经博主允许不得转载. Android Studio编译大工程报错:java.exe'' finished with non-zero exit value 1

[HEOI2014]大工程

题目描述 国家有一个大工程,要给一个非常大的交通网络里建一些新的通道. 我们这个国家位置非常特殊,可以看成是一个单位边权的树,城市位于顶点上. 在 2 个国家 a,b 之间建一条新通道需要的代价为树上 a,b 的最短路径. 现在国家有很多个计划,每个计划都是这样,我们选中了 k 个点,然后在它们两两之间 新建 C(k,2)条 新通道.现在对于每个计划,我们想知道: 1.这些新通道的代价和 2.这些新通道中代价最小的是多少 3.这些新通道中代价最大的是多少 输入输出格式 输入格式: 第一行 n 表