公元 2044 年,人类进入了宇宙纪元。
L 国有 n 个星球,还有 n−1 条双向航道,每条航道建立在两个星球之间,这 n−1 条航道连通了 L 国的所有星球。
小 P 掌管一家物流公司, 该公司有很多个运输计划,每个运输计划形如:有一艘物流飞船需要从 ui 号星球沿最快的宇航路径飞行到 vi 号星球去。显然,飞船驶过一条航道是需要时间的,对于航道 j,任意飞船驶过它所花费的时间为 tj,并且任意两艘飞船之间不会产生任何干扰。
为了鼓励科技创新, L 国国王同意小 P 的物流公司参与 L 国的航道建设,即允许小P 把某一条航道改造成虫洞,飞船驶过虫洞不消耗时间。
在虫洞的建设完成前小 P 的物流公司就预接了 m 个运输计划。在虫洞建设完成后,这 m 个运输计划会同时开始,所有飞船一起出发。当这 m 个运输计划都完成时,小 P 的物流公司的阶段性工作就完成了。
如果小 P 可以自由选择将哪一条航道改造成虫洞, 试求出小 P 的物流公司完成阶段性工作所需要的最短时间是多少?
输入格式
第一行包括两个正整数 n,m,表示 L 国中星球的数量及小 P 公司预接的运输计划的数量,星球从 1 到 n 编号。
接下来 n−1 行描述航道的建设情况,其中第 i 行包含三个整数 ai,bi 和 ti,表示第 i 条双向航道修建在 ai 与 bi 两个星球之间,任意飞船驶过它所花费的时间为 ti。数据保证 1≤ai,bi≤n 且 0≤ti≤1000。
接下来 m 行描述运输计划的情况,其中第 j 行包含两个正整数 uj 和 vj,表示第 j 个运输计划是从 uj 号星球飞往 vj号星球。数据保证 1≤ui,vi≤n
输出格式
输出文件只包含一个整数,表示小 P 的物流公司完成阶段性工作所需要的最短时间。
样例一
input
6 3
1 2 3
1 6 4
3 1 7
4 3 6
3 5 5
3 6
2 5
4 5
output
11
思路:
这题在考场上,我一开始就看错了题目,以为是求扣去一条路让这些路的和最短,于是我就优先打了一遍树链剖分,然而输样例的时候,就有点不对了,我就有点慌了。
然后我细细观察题目发现了这道题是求让最大路最短的这样的问题,这样的问题让我联想到了二分,于是我在考场上就在原来树链剖分的情况下加了个二分(作死的开始,为什么?因为我打错了一些地方XD)。
二分答案值,然后用树链剖分把比答案大的路在树上覆盖,在线段树上找到让这些值合法且被所有路径覆盖的一条路,有则可以,无则不行,很明显可以推算出这是由单调性的,于是我毅然决然写了下去(时间太大,应该能90-95的样子)
95分代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 400000
struct xx{
int l,r,len;
}d[N];
int fa[N],top[N],siz[N],size,son[N],pos[N],n,m,deep[N];
int L[N],R[N],qz[N],tou[N],dd,tree[N*4],tree1[N*4],tree2[N*4];
int sc[N],ans;
struct aa{
int p,next;
}da[N*2];
void dfs(int x){
for (int i=tou[x];i;i=da[i].next){
int v=da[i].p;
if (v!=fa[x]){
fa[v]=x;
deep[v]=deep[x]+1;
dfs(v);
siz[x]+=siz[v];
if (siz[v]>siz[son[x]])son[x]=v;
}
}
siz[x]+=1;
}
void dfs1(int x,int tp){
pos[x]=++size;top[x]=tp;
if (son[x])dfs1(son[x],tp);
for (int i=tou[x];i;i=da[i].next){
int v=da[i].p;
if (v!=fa[x]&&v!=son[x])dfs1(v,v);
}
}
void add(int x,int y){
da[++dd].p=y;da[dd].next=tou[x];tou[x]=dd;
}
void clac(int x){
tree[x]=tree[x*2]+tree[x*2+1];
}
void ins(int x,int l,int r,int xx,int xi,int v){
if (l==r){
tree[x]=v;
return;
}
int mid=(l+r)/2;
if (xx<=mid)ins(x*2,l,mid,xx,xi,v);
if (mid<xx)ins(x*2+1,mid+1,r,xx,xi,v);
clac(x);
}
int query(int x,int l,int r,int ls,int rs){
if (l==r)return tree[x];
if (ls<=l&&r<=rs)return tree[x];
int mid=(l+r)/2;
int mm=0;
if (ls<=mid)mm+=query(x*2,l,mid,ls,rs);
if (mid<rs)mm+=query(x*2+1,mid+1,r,ls,rs);
return mm;
}
int work(int x,int y){
int mm=0;
for (;top[x]!=top[y];x=fa[top[x]]){
if (deep[top[x]]<deep[top[y]])std::swap(x,y);
mm+=query(1,1,n,pos[top[x]],pos[x]);
}
if (pos[x]>pos[y])std::swap(x,y);
if (pos[x]+1<=pos[y]){
mm+=query(1,1,n,pos[x]+1,pos[y]);
}
return mm;
}
void pushdown(int x){
if (tree1[x]){
tree1[x*2]+=tree1[x];
tree1[x*2+1]+=tree1[x];
tree2[x*2]+=tree1[x];
tree2[x*2+1]+=tree1[x];
tree1[x]=0;
}
}
void add1(int x,int l,int r,int ls,int rs,int v){
if (l==r){
tree1[x]+=v;
tree2[x]+=v;
return ;
}
pushdown(x);
if (ls<=l&&r<=rs){
tree1[x]+=v;
tree2[x]+=v;
return;
}
int mid=(l+r)/2;
if (ls<=mid)add1(x*2,l,mid,ls,rs,v);
if (mid<rs)add1(x*2+1,mid+1,r,ls,rs,v);
tree2[x]=std::max(tree2[x*2],tree2[x*2+1]);
}
int work1(int x,int y){//树链剖分
int mm=0;
for (;top[x]!=top[y];x=fa[top[x]]){
if (deep[top[x]]<deep[top[y]])std::swap(x,y);
add1(1,1,n,pos[top[x]],pos[x],1);
}
if (pos[x]>pos[y])std::swap(x,y);
if (pos[x]+1<=pos[y]){
add1(1,1,n,pos[x]+1,pos[y],1);
}
return mm;
}
bool find(int x,int l,int r,int zx,int rx){//线段树上二分查找,找到合法边
if (l==r)
{if (tree[x]>=rx&&tree2[x]==zx)return 1;else return 0;}
pushdown(x);
bool b=0;
int mi=(l+r)/2;
if (tree2[x*2]>=zx)b=b|find(x*2,l,mi,zx,rx);
if (tree2[x*2+1]>=zx)b=b|find(x*2+1,mi+1,r,zx,rx);
return b;
}
bool check(int x){
int ll=1;int rr=m;
for(;ll<=n;ll++)
if (d[ll].len>x)break;
memset(tree1,0,sizeof(tree1));
memset(tree2,0,sizeof(tree2));
while (d[ll].len<=x)++ll;
for (int i=ll;i<=m;i++)work1(d[i].l,d[i].r);
return find(1,1,n,m-ll+1,d[m].len-x);
}
inline bool cmp(const xx &a,const xx &b){return a.len<b.len;}
int main(){
scanf("%d%d",&n,&m);
for (int i=1;i<n;i++){
scanf("%d%d%d",&L[i],&R[i],&qz[i]);
add(L[i],R[i]);
add(R[i],L[i]);
}
dfs(1);
dfs1(1,1);
for (int i=1;i<n;i++){
if (fa[L[i]]==R[i])ins(1,1,n,pos[L[i]],L[i],qz[i]);
else ins(1,1,n,pos[R[i]],R[i],qz[i]);
}
for (int i=1;i<=m;i++){
int x,y;
scanf("%d%d",&x,&y);
d[i].len=work(x,y);
d[i].l=x;d[i].r=y;
}
std::sort(d+1,d+m+1,cmp);
int l=1;int r=d[m].len;
while (l<=r){
int mid=(l+r)/2;
if (check(mid)){
ans=mid;
r=mid-1;
}else l=mid+1;
}
printf("%d",ans);
return 0;
}
考完之后,wking和我探讨了这道题,他发现线段树可以改成差分(都可以解决线段覆盖问题),原本我nlog3(n)的方法被优化的一个log.
100分代码(转自:wkingG博客)
#include<cstdio>
#include<algorithm>
#include<cstring>
const int maxn=664144;
struct range {
int x,y,len;
}d[maxn];
bool cmp(range x,range y){return x.len>y.len;}
int root,n,m,x,y,tot,next[maxn],last[maxn],e[maxn],dep[maxn],fa[maxn],size[maxn],son[maxn],top[maxn],pos[maxn];
int pos2[maxn],val[maxn],cost[maxn],MAX[maxn],dist[maxn],s[maxn],node,ans,q[maxn];
int read()//优化之一
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchr();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void add(int x,int y,int v)
{
next[++tot]=last[x];
last[x]=tot;
val[tot]=v;
e[tot]=y;
}
void dfs1(int x)
{
dep[x]=dep[fa[x]]+1;
size[x]=1;
for (int i=last[x];i;i=next[i])
{
int v=e[i];
if (v==fa[x])continue;
fa[v]=x;
dist[v]=dist[x]+val[i];
cost[v]=val[i];
dfs1(v);
size[x]+=size[v];
if (size[v]>size[son[x]])son[x]=v;
}
}
void dfs2(int x,int tp)//树链剖分
{
top[x]=tp;pos[x]=++node;pos2[node]=cost[x];
if (son[x]==0)return;
dfs2(son[x],tp);
for (int i=last[x];i;i=next[i])
{
int v=e[i];
if (v==fa[x]||v==son[x])continue;
dfs2(v,v);
}
}
int query(int x,int y)//lca
{
int a=x,b=y;
while (top[a]!=top[b])
{
if (dep[top[a]]<dep[top[b]])std::swap(a,b);
a=fa[top[a]];
}
if (dep[a]<dep[b])return a;else return b;
}
void work(int x,int y)//差分
{
int a=x,b=y;
while (top[a]!=top[b])
{
if (dep[top[a]]<dep[top[b]])std::swap(a,b);
s[pos[top[a]]]+=1;s[pos[a]+1]-=1;
a=fa[top[a]];
}
if (a==b)return;
if (dep[a]>dep[b]) std::swap(a,b);
s[pos[a]+1]+=1;s[pos[b]+1]-=1;;
}
int check(int mid)
{
int sz=0;
while (d[sz+1].len>mid)sz++;
if (q[sz]!=0)return q[sz];//优化之一 记忆化
std::memset(s,0,sizeof(s));
for (int i=1;i<=sz;i++)work(d[i].x,d[i].y);
int Maxx=0,tott=0;
for (int i=1;i<=n;i++){tott+=s[i];if (tott==(sz))Maxx=std::max(Maxx,pos2[i]);}
q[sz]=Maxx;
return Maxx;
}
int main()
{
n=read();m=read();
for (int i=1;i<n;i++)
{
int v;
x=read();y=read();v=read();
add(y,x,v);add(x,y,v);
}
dfs1(1);
dfs2(1,1);
for (int i=1;i<=m;i++)
{
d[i].x=read();d[i].y=read();
d[i].len=dist[d[i].x]+dist[d[i].y]-2*dist[query(d[i].x,d[i].y)];
}
std::sort(d+1,d+1+m,cmp);
int l=0,r=d[1].len;
while (l<=r)
{
int mid=(l+r)>>1;
if (d[1].len-check(mid)>mid)
{
l=mid+1;
}else r=mid-1,ans=mid;
}
printf("%d",ans);
return 0;
}