bzoj 3872 [ Poi 2014 ] Ant colony —— 二分

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3872

从食蚁兽所在的边向叶节点推，会得到一个渐渐放大的取值区间，在叶子节点上二分有几群蚂蚁符合条件即可；

注意中途判断，如果已经超过范围就返回或者处理一下，据说会爆 long long 之类的；

而且食蚁兽所在的边的两个端点的初始值不一定是 k 和 k+1 ！也要看度数！

注意统计答案的 num 也是 long long 。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mid ((ll+rr)>>1)
using namespace std;
typedef long long ll;
int const xn=1e6+;
int n,g,k,s,t,hd[xn],ct,to[xn<<],nxt[xn<<],deg[xn];
ll m[xn],l[xn],r[xn],num;//
bool vis[xn];
int rd()
{
    int ret=,f=; char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=; ch=getchar();}
    while(ch>=''&&ch<='')ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
    return f?ret:-ret;
}
void add(int x,int y){to[++ct]=y; nxt[ct]=hd[x]; hd[x]=ct;}
int findl(ll x)//第一个大于等于x的
{
    int ll=,rr=g,ret=-;
    while(ll<=rr)
    {
        if(m[mid]>=x)ret=mid,rr=mid-;
        else ll=mid+;
    }
    return ret;
}
int findr(ll x)
{
    int ll=,rr=g,ret=-;
    while(ll<=rr)
    {
        if(m[mid]<x)ret=mid,ll=mid+;//<
        else rr=mid-;
    }
    return ret;
}
void cal(int x)
{
    if(l[x]>m[g]||r[x]<=m[])return;
    num+=findr(r[x])-findl(l[x])+;
}
void dfs(int x)
{
    vis[x]=;
    if(l[x]>m[g])return;//
    if(r[x]>m[g])r[x]=m[g]+;//
    for(int i=hd[x],u;i;i=nxt[i])
    {
        if(vis[u=to[i]]||(x==s&&u==t)||(x==t&&u==s))continue;
        if(deg[u]>)
        {
            l[u]=(deg[u]-)*l[x]; r[u]=(deg[u]-)*r[x];
            dfs(u);
        }
        else cal(x);
    }
}
int main()
{
    n=rd(); g=rd(); k=rd();
    for(int i=;i<=g;i++)m[i]=rd();
    sort(m+,m+g+);
    for(int i=,x,y;i<n;i++)
    {
        x=rd(); y=rd();
        if(i==)s=x,t=y;
        add(x,y); add(y,x); deg[x]++; deg[y]++;
    }
//    l[s]=l[t]=k; r[s]=r[t]=k+1;
    if(deg[s]==)l[s]=k,r[s]=k+;
    else l[s]=(deg[s]-)*k,r[s]=(deg[s]-)*(k+);
    if(deg[s]==)cal(s);//!
    if(deg[t]==)l[t]=k,r[t]=k+;
    else l[t]=(deg[t]-)*k,r[t]=(deg[t]-)*(k+);
    if(deg[t]==)cal(t);//!
    dfs(s); dfs(t);
    printf("%lld\n",num*k);
    return ;
}