设为首页 加入收藏

TOP

BZOJ 3611 HEOI2014 大工程 倍增LCA+单调栈+树形DP
2015-07-20 17:25:27 来源: 作者: 【 】 浏览:3
Tags:BZOJ 3611 HEOI2014 工程 倍增 LCA 单调 树形

题目大意:给定一棵树,m次询问,每次给出k个关键点,询问这k个点之间的两两距离和、最小距离和最大距离

n<=100W,m<=50000,Σk<=2*n

?

\

\

处理方法同2286 消耗战 地址见 www.2cto.com

这个题的DP有些麻烦 因此我把要处理的节点单独拎出来做的DP 具体状态和转移见代码

?

#include 
  
   
#include 
   
     #include 
    
      #include 
     
       #define M 1001001 #define INF 0x3f3f3f3f using namespace std; struct abcd{ int to,next; }table[M<<1]; int head[M],tot; int n,m; int pos[M],dpt[M],fa[M][20]; long long ans,ans_min,ans_max; bool is_key_point[M]; void Add(int x,int y) { table[++tot].to=y; table[tot].next=head[x]; head[x]=tot; } void DFS(int x) { static int cnt=0; int i; pos[x]=++cnt;dpt[x]=dpt[fa[x][0]]+1; for(i=head[x];i;i=table[i].next) if(table[i].to!=fa[x][0]) { fa[table[i].to][0]=x; DFS(table[i].to); } } int LCA(int x,int y) { int j; if(dpt[x]
      
       =dpt[y]) x=fa[x][j]; if(x==y) return x; for(j=19;~j;j--) if(fa[x][j]!=fa[y][j]) x=fa[x][j],y=fa[y][j]; return fa[x][0]; } bool Compare(int x,int y) { return pos[x] < pos[y] ; } void Tree_DP(int x,int from) { static long long f[M],g[M],max_dis[M],min_dis[M]; //f[x]表示以x为根的子树中有多少关键点 //g[x]表示以x为根的子树中所有关键点到x的距离之和 //max_dis[x]/min_dis[x]表示节点x为根的子树中的关键点到x的距离的最大/最小值 int i; f[x]=is_key_point[x];g[x]=0; max_dis[x]=(is_key_point[x]?0:-INF); min_dis[x]=(is_key_point[x]?0:INF); for(i=head[x];i;i=table[i].next) { if(table[i].to==from) continue; Tree_DP(table[i].to,x); int dis=dpt[table[i].to]-dpt[x]; ans+=(g[x]+f[x]*dis)*f[table[i].to]+g[table[i].to]*f[x]; ans_min=min(ans_min,min_dis[x]+min_dis[table[i].to]+dis); ans_max=max(ans_max,max_dis[x]+max_dis[table[i].to]+dis); f[x]+=f[table[i].to]; g[x]+=g[table[i].to]+f[table[i].to]*dis; max_dis[x]=max(max_dis[x],max_dis[table[i].to]+dis); min_dis[x]=min(min_dis[x],min_dis[table[i].to]+dis); } } int main() { int i,j,k,x,y; cin>>n; for(i=1;i
       
        >m; for(i=1;i<=m;i++) { static int a[M]; static int stack[M],top; scanf(%d,&k); for(j=1;j<=k;j++) scanf(%d,&a[j]); sort(a+1,a+k+1,Compare); tot=0; stack[top=1]=1; head[1]=0;is_key_point[1]=(a[1]==1); for(j=1;j<=k;j++) { int lca=LCA(a[j],stack[top]); while(dpt[lca]
        
         1) Add(stack[top-1],stack[top]),top--; ans=0;ans_min=INF;ans_max=-INF; Tree_DP(1,0); #ifdef ONLINE_JUDGE printf(%lld %lld %lld ,ans,ans_min,ans_max); #else printf(%I64d %I64d %I64d ,ans,ans_min,ans_max); #endif } return 0; } 
        
       
      
     
    
   
  


?

】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
分享到: 
上一篇NYOJ 420 p次方求和 (快速幂+同.. 下一篇NYOJ 285 寻找克隆人(map+计数)

评论

帐  号: 密码: (新用户注册)
验 证 码:
表  情:
内  容:

·C 内存管理 | 菜鸟教 (2025-12-26 20:20:37)
·如何在 C 语言函数中 (2025-12-26 20:20:34)
·国际音标 [ç] (2025-12-26 20:20:31)
·微服务 Spring Boot (2025-12-26 18:20:10)
·如何调整 Redis 内存 (2025-12-26 18:20:07)