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;
}
版权声明: ˋ( ° ▽、° )  ̄へ ̄

相关文章推荐

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

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

LintCode-二叉树中的最大路径和

给出一棵二叉树,寻找一条路径使其路径和最大,路径可以在任一节点中开始和结束(路径和为两个节点之间所在路径上的节点权值之和) 样例 给出一棵二叉树: 1 / ...

Hdu 5664 Lady CA and the graph(有n个点的树,给定根,叫你找第k大的特殊链)

传送门:Hdu 5664 Lady CA and the graph 题意: 给你一个有n个点的树,给定根,叫你找第k大的特殊链 特殊的链的定义:u,v之间的路径,且lca(u,v)!=...

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

传送门:【HDU】5314 Happy King算法分析: 点分治,子树去重类可做。对于一个重心,取出所有路径按照最大值排序,扫下最小值,插入到线段树里维护个数,双指针扫扫解决。然后对所有重心的儿子...

hdu-5314 Happy King

题意: 给出一颗n个结点的树,点上有权值; 求点对(x,y)满足x!=y且x到y的路径上最大值与最小值的差 n 题解: 来填一填当年挖下的坑; 这个数据范围真是恶意。。直接说五组数...

HDU 5314 Happy King 点分治

/* HDU 5314 题意: 给定N个点的树,点有权值 求多少个点对(u,v)满足u到v的路径上点权值最大值减最小值不大于给定的K 分治瞎搞。...

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

Happy King Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 262144/262144 K (Java/Others) To...

Bzoj 3672 购票(树分治+凸壳维护)

题意:给出一棵有根树(1为根),边有长度。每个点u有三个属性(len[u],p[u],q[u]),每次u可以转移到u的某个祖先节点v(v满足dist(u,v)...

Kattis yatp(斜率优化+树分治)

题意: 2e5个点的无根树,每个点有点权,每条边有边权,定义一条简单路径的花费=这条路径两个端点点权的乘积+边权和, (一条简单路径可以包含一个点,这样花费是该点权的平方),最后问从每个点...

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

1.区间是一段的,不是断开的哟 2.代码是看着标程写的 3.枚举左端点,二分右端点流程: #include #include #include #define LL long l...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Hdu 5314 Happy King(求树上多少个点对(u,v)满足u到v的路径上点权值最大值减最小值不大于给定的K)
举报原因:
原因补充:

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