Hdu 5314 Happy King(求树上多少个点对(u,v)满足u到v的路径上点权值最大值减最小值不大于给定的K)

原创 2016年08月28日 15:42:06

传送门:Hdu 5314 Happy King


题意:
给定N个点的树,点有权值,求多少个点对(u,v)满足u到v的路径上点权值最大值减最小值不大于给定的K


思路:
将点对分成经过根的和不经过根的,进行分治
每一次分治都维护点到根的最大值和最小值就可以了,
处理完一个子树的时候,利用容斥+二分查找左右范围就可以了
(路径上两点的较小值是两点到根较小的那个,较大值为两点到根较大的)


#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <ctime>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
using namespace std;
#define INF 0x3f3f3f3f
#define inf -0x3f3f3f3f
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define mem(a, b) memset(a, b, sizeof(a))
#define MP(x,y) make_pair(x,y)
typedef long long ll;
void fre() { freopen("input.in", "r", stdin); freopen("output.out", "w", stdout); }
template <class T1, class T2>inline void gmax(T1 &a, T2 b) { if (b>a)a = b; }
template <class T1, class T2>inline void gmin(T1 &a, T2 b) { if (b<a)a = b; }
typedef pair<int,int>PI;
const int maxn=100100;
const int MAXM=2*maxn;
vector<int>G[maxn];
int dis[maxn],k,size[maxn],f[maxn],Count,root,a[maxn];//Count表示当前子树的结点的总个数
bool Del[maxn];
long long ans=0;
struct node{
    int _min,_max;
    bool operator < (const node& a)  const{
        return a._min==_min ? _max<a._max : _min<a._min;
    }
}P[maxn];
int C[maxn],cnt;

struct Edge{
    int to,next;
}e[MAXM];
int tot,head[maxn];

void init(){
    tot=0;
    memset(head,-1,sizeof(head));
}

void addedge(int u,int v){
    e[tot].to=v;
    e[tot].next=head[u];
    head[u]=tot++;
}

void getroot(int u,int pre){
    size[u]=1,f[u]=0;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v!=pre && !Del[v]){
            getroot(v,u);
            size[u]+=size[v];
            f[u]=max(f[u],size[v]);
        }
    }
    f[u]=max(f[u],Count-size[u]);
    if(f[u]<f[root])    root=u;
}

void getdep(int u,int pre,int _min,int _max){ //这里还需要重新计算每个子树的size
    _min=min(a[u],_min),_max=max(a[u],_max);
    if(_min+k>=_max)
        P[++cnt]=(node){_min,_max};
    size[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v!=pre && !Del[v]){
            getdep(v,u,_min,_max);
            size[u]+=size[v];
        }
    }
}

void add(int x,int add,int m){
    while(x<=m)
        C[x]+=add,x+=(x&-x);
}

int sum(int x){
    int ret=0;
    while(x>0)
        ret+=C[x],x-=(x&-x);
    return ret;
}

long long cal(int u,int _min,int _max){
    cnt=0;
    getdep(u,0,_min,_max);
    sort(P+1,P+cnt+1);
    long long ret=0;
    for(int i=cnt;i>=1;i--){    //最小值为P[i]._min,最大值大于等于P[i]._min,小于等于P[i]._min+k
        int num=lower_bound(P+1,P+i,(node){P[i]._max-k,0})-P; //后面的最大值一定比P[i]._minv大
        ret+=(i-num);
    }
    return ret;
}

void work(int u){
    ans+=cal(u,INF,-INF);
    Del[u]=true;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(!Del[v]){
            ans-=cal(v,a[u],a[u]);
            f[0]=Count=size[v];
            getroot(v,root=0);
            work(root);
        }
    }
}

int main(){
    int n,_;
    scanf("%d",&_);
    while(_--){
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        init();
        int u,v,w;
        for(int i=1;i<n;i++){
            scanf("%d%d",&u,&v);
            addedge(u,v),addedge(v,u);
        }
        ans=0;
        f[0]=Count=n;
        getroot(1,root=0);
        memset(Del,false,sizeof(Del));
        work(root);
        printf("%lld\n",ans*2);
    }
    return 0;
}
版权声明: ˋ( ° ▽、° )  ̄へ ̄

【HDU3530】【单调队列(双)】Subsequence 【长度为n的数列,求最长子区间的长度,使得区间的最大值与最小值的差满足一个范围】

传送门:HDU 3530 Subsequence 描述: Subsequence Time Limit: 2000/1000 MS (Java/Others)    Memory Lim...
  • guhaiteng
  • guhaiteng
  • 2016年10月08日 15:57
  • 1308

分治算法在树的路径问题上的应用

分治算法在树的路径问题中的应用 【摘要】          树作为一类特殊的数据结构,在信息学中有着极为重要的作用,各类关于树的题目在竞赛中更是屡见不鲜。本文选取了近几年出现的关于树的路径的题...
  • Taceo
  • Taceo
  • 2013年08月12日 21:43
  • 1781

最短路径算法--Dijkstra算法,Bellmanford算法,Floyd算法,Johnson算法

大数据技术虫 最短路径算法 在交通地图上,两地点之间的路径通常标有长度,我们可以用加权有向来描述地图上的交通网。加权有向图中每条路径都有一个路径权值...
  • u011089523
  • u011089523
  • 2015年12月07日 10:04
  • 6142

hdu 5289 Assignment(给一个数组,求有多少个区间,满足区间内的最大值和最小值之差小于k)

1.区间是一段的,不是断开的哟 2.代码是看着标程写的 3.枚举左端点,二分右端点流程: #include #include #include #define LL long l...
  • xky1306102chenhong
  • xky1306102chenhong
  • 2015年07月22日 12:53
  • 1519

Hdu 4918 Query on the subtree(一棵树,两种操作,一种是将某个点的权值修改为v,另一种是查询距离点u不超过d的点的权值和。)

传送门:Hdu 4918 Query on the subtree 题意:给出一颗n个点的树,每个点有一个权值,有两种操作,一种是将某个点的权值修改为v,另一种是查询距离点u不超过d的点的权值和。 ...
  • acm_fighting
  • acm_fighting
  • 2016年08月30日 20:59
  • 353

求一个数组中第K个最大值和最小值

  • 2006年02月23日 09:05
  • 1.8MB
  • 下载

hdu 5314 Happy King 树点分冶 树状数组

Happy King Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 262144/262144 K (Java/Others) To...
  • u010422038
  • u010422038
  • 2015年07月31日 22:49
  • 752

Hdu 5866 Lucky E(树上点期望的最大值)

传送门:Hdu 5866 Lucky E 题意:给你一棵树,有n个节点,m种操作,树上的每个点相当于一个士兵,有一个伤值, m中操作,每种操作都有一个点的标号和一个概率,表示可以对这个点的子树...
  • acm_fighting
  • acm_fighting
  • 2016年08月20日 10:08
  • 443

HDU 5242Game 树上的贪心 树形dp 求出使K条链的权值总和最大

树链剖分找权值最大的前k条链 题目大意:给定一个树形的游戏网络,可以从根节点出发k个人,每个人可以沿着一条路径走下去,不能回头,出口在各个叶子节点,在路过一个节点时可以           获得该点的...
  • became_a_wolf
  • became_a_wolf
  • 2016年05月21日 22:05
  • 441

【HDU】5314 Happy King【动态树(点分治)】

传送门:【HDU】5314 Happy King算法分析: 点分治,子树去重类可做。对于一个重心,取出所有路径按照最大值排序,扫下最小值,插入到线段树里维护个数,双指针扫扫解决。然后对所有重心的儿子...
  • u013368721
  • u013368721
  • 2015年07月27日 16:29
  • 1203
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Hdu 5314 Happy King(求树上多少个点对(u,v)满足u到v的路径上点权值最大值减最小值不大于给定的K)
举报原因:
原因补充:

(最多只允许输入30个字)