BZOJ3872 : [Poi2014]Ant colony

设i点的度数为d[i]

则如果有x只蚂蚁在从i走到别处，会分裂成每群$\lfloor\frac{x}{d[i]-1}\rfloor$只蚂蚁

对于x出发的m只蚂蚁，到y处还剩$\lfloor\frac{m}{x到y路径上所有点度数-1的乘积}\rfloor$

于是我们一遍DFS求出到每个叶子节点时一路上点度数-1的乘积，再在数组中二分查找出一个合法的区间即可

 

#include<cstdio>
#include<algorithm>
#define N 1000010
typedef long long ll;
int n,m,k,i,j,x,y,g[N],nxt[N<<1],v[N<<1],d[N],ed,a[N];ll ans;
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline void add(int x,int y){d[x]++;v[++ed]=y;nxt[ed]=g[x];g[x]=ed;}
inline int ask(ll x){
  if(x>a[m])return m;
  if(x<=a[1])return 0;
  int l=1,r=m,t,mid;
  while(l<=r)if(a[mid=(l+r)>>1]<x)l=(t=mid)+1;else r=mid-1;
  return t;
}
void dfs(int x,int y,int t){
  if(!d[x])ans+=ask((ll)(k+1)*t)-ask((ll)k*t);
  if((ll)t*d[x]>a[m])return;t*=d[x];
  for(int i=g[x];i;i=nxt[i])if(i>2&&v[i]!=y)dfs(v[i],x,t);
}
int main(){
  read(n),read(m),read(k);
  for(i=1;i<=m;i++)read(a[i]);
  std::sort(a+1,a+m+1);
  for(i=1;i<n;i++)read(x),read(y),add(x,y),add(y,x);
  for(i=1;i<=n;i++)d[i]--;
  dfs(v[1],0,1);dfs(v[2],0,1);
  return printf("%lld",ans*k),0;
}