题目:
题意:
找出一条长度[l,r]的中位数最大的路径。
题解:
还是使用二分,对于每条边的权值,如果>=mid就是1,< mid就是-1,如果一条路径的权值和>=0,证明这个中位数是合法的
别的就和昨天那个是一样的了?
但是我交上去是WA,网上找的AC代码配合下载下来的数据都是RE。。。。?
具体原因。。。。
代码:
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int N=100005;
const int INF=1e9;
int tot,nxt[N*2],point[N],v[N*2],c[N*2],L,R,mid,root,f[N],size[N],deep[N],sum,DEP,dep,a[N],b[N],q[N],ans,s,xin[N];bool vis[N];
void addline(int x,int y,int z)
{
++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
void getroot(int x,int fa)
{
f[x]=0; size[x]=1;
for (int i=point[x];i;i=nxt[i])
if (v[i]!=fa && !vis[v[i]])
{
getroot(v[i],x);
size[x]+=size[v[i]];
f[x]=max(f[x],size[v[i]]);
}
f[x]=max(f[x],sum-size[x]);
if (f[x]<f[root]) root=x;
}
void getdeep(int x,int fa)
{
deep[x]=deep[fa]+1;
for (int i=point[x];i;i=nxt[i])
if (!vis[v[i]] && v[i]!=fa) getdeep(v[i],x);
DEP=max(DEP,deep[x]);
}
void dfs(int x,int fa,int ok)
{
if (deep[x]>R) return;
a[deep[x]]=max(a[deep[x]],ok);
for (int i=point[x];i;i=nxt[i])
if (!vis[v[i]] && v[i]!=fa) dfs(v[i],x,ok+((c[i]>=mid)?1:-1));
dep=max(dep,deep[x]);
}
bool check()
{
int maxx=-INF,now=0;
for (int i=1;i<=DEP;i++) b[i]=-INF;
for (int i=point[root];i;i=nxt[i])
if (!vis[v[i]])
{
for (int j=1;j<=DEP;j++) a[j]=-INF;
dep=0;int l=1,r=0;
dfs(v[i],root,(c[i]>=mid)?1:-1);
if (now) q[++r]=now;
for (int j=1;j<=dep;j++)
{
if (j>R) break;
while (l<=r && q[l]+j>R) l++;
if (j<=L && L-j<=DEP)
{
while (l<=r && b[q[r]]<b[L-j]) r--;
q[++r]=L-j;
maxx=max(maxx,b[q[l]]+a[j]);
}
else if (j>L) maxx=max(maxx,b[q[l]]+a[j]);
}
for (int j=1;j<=dep;j++)
{
b[j]=max(b[j],a[j]);
if (j>=L && j<=R && b[now]<b[j]) now=j;
}
}
return maxx>=0;
}
void work(int x)
{
DEP=0;getdeep(root,0);
int l=ans,r=s;
while (l<=r)
{
int Mid=(l+r)>>1; mid=xin[Mid];
if (check()) l=Mid+1,ans=Mid;else r=Mid-1;
}
vis[x]=1;
for (int i=point[x];i;i=nxt[i])
if (!vis[v[i]])
{
sum=size[v[i]]; if (sum<L) continue;
root=0; f[0]=INF; getroot(v[i],0);
work(root);
}
}
int main()
{
int n;scanf("%d%d%d",&n,&L,&R);
for (int i=1;i<n;i++)
{
int x,y,z;scanf("%d%d%d",&x,&y,&z);
addline(x,y,z);xin[i]=z;
}
sort(xin+1,xin+n);s=unique(xin+1,xin+n)-xin-1;
xin[0]=-1;
ans=0;deep[0]=-1;
sum=n; root=0; f[0]=INF; getroot(1,0);
work(root);
printf("%d",xin[ans]);
}