bzoj 3872 [Poi2014]Ant colony——二分答案

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

可以倒推出每个叶子节点可以接受的值域。然后每个叶子二分有多少个区间符合即可。

注意一开始的两个点不是直接是 l[ u ]=r[ u ]=lm ！也要看度数的！且把那条边的两个端点分别算子树很方便。

而且过程中似乎会爆 long long ，所以如果 l[ i ] 都大于最大值就不往下算了；r[ i ]也要每次与最大值取min。

然后是和网上一份题解对拍出错却仍A了此题的代码。不想管是哪错了……

（连freopen都忘了去掉也能A！）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e6+;
int n,m,lm,a[N],hd[N],xnt,to[N<<],nxt[N<<];
int rt1,rt2,deg[N];
ll l[N],r[N],ans;
int rdn()
{
    int ret=;bool fx=;char ch=getchar();
    while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
    while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
    return fx?ret:-ret;
}
void add(int x,int y)
{
    to[++xnt]=y; nxt[xnt]=hd[x]; hd[x]=xnt;
    to[++xnt]=x; nxt[xnt]=hd[y]; hd[y]=xnt;
    deg[x]++; deg[y]++;
}
void dfs(int cr,int fa)
{
    //printf("l[%d]=%lld r[%d]=%lld\n",cr,l[cr],cr,r[cr]);
    for(int i=hd[cr],v;i;i=nxt[i])
        if(i>&&(v=to[i])!=fa)
        {
            if(deg[v]==)
            {
                l[v]=l[cr]; r[v]=r[cr];
            }
            else
            {
                l[v]=l[cr]*(deg[v]-);
                r[v]=r[cr]*(deg[v]-)+deg[v]-;
            }
            r[v]=min(r[v],(ll)a[m]);///
            if(l[v]>a[m])continue;///
            dfs(v,cr);
        }
}
int findr(ll x)//ll
{
    int l=,r=m,ret=;
    while(l<=r)
    {
        int mid=l+r>>;
        if(a[mid]<=x)ret=mid,l=mid+;
        else r=mid-;
    }
    return ret;
}
int findl(ll x)
{
    int l=,r=m,ret=m+;
    while(l<=r)
    {
        int mid=l+r>>;
        if(a[mid]>=x)ret=mid,r=mid-;
        else l=mid+;
    }
    return ret;
}
int main()
{
    freopen("data.in","r",stdin);
    n=rdn(); m=rdn(); lm=rdn();
    for(int i=;i<=m;i++) a[i]=rdn();
    sort(a+,a+m+);

    int u,v; u=rdn(); v=rdn();
    rt1=u; rt2=v; add(u,v);
//    l[u]=r[u]=l[v]=r[v]=lm;
    for(int i=;i<n;i++)
    {
        u=rdn(); v=rdn(); add(u,v);
    }

    u=rt1; v=rt2;
    if(deg[u]==) l[u]=r[u]=lm;/////
    else l[u]=lm*(deg[u]-),r[u]=lm*(deg[u]-)+deg[u]-;///
    r[u]=min(r[u],(ll)a[m]); if(l[u]<=a[m]) dfs(u,);

    if(deg[v]==) l[v]=r[v]=lm;
    else l[v]=lm*(deg[v]-),r[v]=lm*(deg[v]-)+deg[v]-;
    r[v]=min(r[v],(ll)a[m]); if(l[v]<=a[m]) dfs(v,);

    for(int i=;i<=n;i++)
        if(deg[i]==&&r[i]&&l[i]<=a[m])
            ans+=findr(r[i])-findl(l[i])+;
    printf("%lld\n",ans*lm);
    return ;
}