题目：https://www.luogu.org/problemnew/show/P1084
5个月前曾经写过一次，某个上学日的深夜，精疲力竭后只有区区10分，从此没管…
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
priority_queue< pair<int,int> >q1,q2;
int const MAXN=;
int n,m,head[MAXN],ct,ans,fa[MAXN],sid[MAXN],lf,lef[MAXN],gr[MAXN];
ll l,r;
bool vis[MAXN],in[MAXN];
struct N{
int to,next,w;
N(int t=,int n=,int w=):to(t),next(n),w(w) {}
}edge[MAXN<<];
struct E{
int st,pos;
ll rest;
}am[MAXN];
void add(int x,int y,int z)
{
edge[++ct]=N(y,head[x],z);head[x]=ct;
edge[++ct]=N(x,head[y],z);head[y]=ct;
}
void ps(int x)
{
int nw=am[x].st;
while(fa[nw]!=&&am[x].rest-sid[nw])
{
am[x].rest-=sid[nw];
nw=fa[nw];
}
if(fa[nw]==&&am[x].rest>=*sid[nw])am[x].pos=,am[x].rest-=sid[nw];
else am[x].pos=nw;
}
void init(int x,int f,int g,int w)//fa,gr,lef,sid
{
int t=;
fa[x]=f;sid[x]=w;
if(g==)gr[x]=x;
else gr[x]=g;
for(int i=head[x],u;i;i=edge[i].next)
{
u=edge[i].to;
if(u==f)continue;
init(u,x,gr[x],edge[i].w);t++;
}
if(!t)lef[++lf]=x;
}
void dfs(int x)
{
vis[x]=;
for(int i=head[x],u;i;i=edge[i].next)
if(edge[i].to!=fa[x])dfs(edge[i].to);
}
bool pd(ll mid)
{
// printf("mid=%d\n",mid);
memset(vis,,sizeof vis);
memset(in,,sizeof in);
while(q1.size())q1.pop();
while(q2.size())q2.pop();
for(int i=;i<=m;i++)
{
am[i].rest=mid;
ps(i);
if(am[i].pos!=)dfs(am[i].pos);
q1.push(make_pair(-am[i].rest,i));
}
// for(int i=1;i<=m;i++)
// printf("am[%d].pos=%d\n",i,am[i].pos);
for(int i=;i<=lf;i++)
if(!vis[lef[i]]&&!in[gr[lef[i]]])
{
q2.push(make_pair(-sid[gr[lef[i]]],gr[lef[i]]));
in[gr[lef[i]]]=;
}
// printf("q1:%d q2:%d\n",q1.size(),q2.size());
while(q1.size()&&q2.size())
{
int x=q1.top().second;q1.pop();//am
int y=q2.top().second;q2.pop();//gr
// printf("x=%d y=%d\n",x,y);
while(am[x].rest<sid[gr[y]]&&q1.size())x=q1.top().second,q1.pop();
if(am[x].rest<sid[gr[y]]&&!q1.size())return ;
} if(q2.size())return ;
else return ;
}
int main()
{
scanf("%d",&n);
for(int i=,x,y,z;i<n;i++)
{
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);r+=z;
}
init(,,,);
// for(int i=1;i<=lf;i++)
// printf("lef[%d]=%d\n",i,lef[i]);
scanf("%d",&m);
for(int i=;i<=m;i++)
scanf("%d",&am[i].st);
ans=-;
while(l<=r)
{
ll mid=((l+r)>>);
// cout<<ans<<endl;
if(pd(mid))ans=mid,r=mid-;
else l=mid+;
}
printf("%d",ans);
return ;
}
囧1
这几天又重新写它了，自信码力已和当时不可同日而语，于是又写了一遍，然而仅仅20分，改了改成了30分…
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
typedef long long ll;
int const xn=;
int n,m,cnt,hd[xn],ct,to[xn<<],nxt[xn<<],w[xn<<],rs[xn],lf[xn],rk[xn];
int f[xn][],son[xn],dep[xn],num[xn];
ll dis[xn],sum,ans;
bool vis[xn],use[xn];
priority_queue<int>q;
vector<int>v[xn];
struct N{
int st,pos; ll tim;
bool operator < (const N &y) const
{return tim<y.tim;}
}a[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,int z){to[++ct]=y; nxt[ct]=hd[x]; w[ct]=z; hd[x]=ct;}
int dfs(int x,int fa,int r)
{
int ret=,fl=; f[x][]=fa;
for(int i=;i<=;i++)f[x][i]=f[f[x][i-]][i-];
for(int i=hd[x],u;i;i=nxt[i])
{
if((u=to[i])==fa)continue;
fl=;
dis[u]=dis[x]+w[i]; dep[u]=dep[x]+;
if(x==)lf[u]+=dfs(u,x,u),son[++cnt]=u;
else ret+=dfs(u,x,r);
}
if(!fl)ret=,rs[x]=r;
return ret;
}
bool cmp(int a,int b){return dis[son[a]]>dis[son[b]];}
void dfs2(int x)
{
bool fl=;
for(int i=hd[x],u;i;i=nxt[i])
if((u=to[i])!=f[x][])fl=,dfs2(u);
if(!fl)num[rs[x]]++;
}
bool ck(int mid)
{
memset(vis,,sizeof vis);
memset(num,,sizeof num);
memset(use,,sizeof use);
for(int i=;i<=cnt;i++)v[i].clear();
while(q.size())q.pop();
for(int i=;i<=m;i++)a[i].pos=;
for(int i=;i<=m;i++)
{
int x=a[i].st; ll ret=mid;
for(int j=;j>=;j--)
if(f[x][j]>&&dis[x]-dis[f[x][j]]<=ret)
ret-=dis[x]-dis[f[x][j]],x=f[x][j];
if(dep[x]>||(dep[x]==&&ret<dis[x]))dfs2(x);
else
{
a[i].pos=x; a[i].tim=ret-dis[x];
q.push(i); //x:1~n
v[x].push_back(i);
// printf("in:%d\n",i);//tim从大到小
}
}
for(int i=;i<=cnt;i++)
if(num[son[i]]==lf[son[i]])vis[i]=;//i != son[i] // vis[1~cnt]
for(int i=;i<=cnt;i++)
{
int x=rk[i];//1~cnt
// printf("vis[%d]=%d\n",son[x],vis[x]);
if(vis[x])continue;
if(!q.size())return ;
int k=q.top(); q.pop();
while(use[k])k=q.top(),q.pop();
if(a[k].tim<dis[son[x]])
{
int us=-;
for(int j=;j<v[son[x]].size();j++)
{
int nw=v[son[x]][j];
if(!use[nw]&&(us==-||a[nw].tim<a[us].tim))us=nw;
}
if(us==-)return ;
else use[us]=,q.push(k);
}
vis[x]=;
// printf("%d -> %d\n",i,son[x]);
}
// printf("return 1\n");
return ;
}
int main()
{
n=rd();
for(int i=,x,y,z;i<n;i++)
{
x=rd(); y=rd(); z=rd();
add(x,y,z); add(y,x,z);
sum+=z;
}
m=rd();
for(int i=;i<=m;i++)a[i].st=rd();
dfs(,,);
for(int i=;i<=cnt;i++)rk[i]=i;
sort(rk+,rk+cnt+,cmp);//rk -> 1~cnt
// for(int i=1;i<=cnt;i++)printf("dis[%d]=%lld rk[%d]=%d\n",son[i],dis[son[i]],i,rk[i]);
ll l=,r=sum; ans=-;
while(l<=r)
{
ll mid=((l+r)>>1ll);
// printf("l=%lld r=%lld mid=%lld\n",l,r,mid);
if(ck(mid))ans=mid,r=mid-;
else l=mid+;
}
printf("%lld\n",ans);
return ;
}
囧2
然后又去参考了TJ…看到思路和我的一样，但是代码简洁很多…
于是改改改，然后和模仿的那篇TJ拍拍拍，居然拍出了TJ的错…又和另一篇TJ拍拍拍，大数据又有错…
不管了直接交上去，就A了…
然后发现数据生成没管军队不在根上，但第一篇TJ小数据真的错了…
和 Narh 的代码拍，大数据又有错…但是数据那么大怎么改…不管了(反正也A了)…
又练习码力了，没事不要多写什么 queue 啦， vector 啦，直接排个序就好了。
代码如下：
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
int const xn=;
int n,m,hd[xn],ct,to[xn<<],nxt[xn<<],w[xn<<],dis[xn],sum,ans;
int f[xn][],num[xn],mn[xn],cnta,cnts,st[xn];
bool vis[xn],use[xn];
struct N{
int id,tim;
bool operator < (const N &y) const
{return tim>y.tim;}
}a[xn],son[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,int z){to[++ct]=y; nxt[ct]=hd[x]; w[ct]=z; hd[x]=ct;}
void dfs(int x,int fa,int r)
{
int ret=; f[x][]=fa;
for(int i=;i<=;i++)f[x][i]=f[f[x][i-]][i-];
for(int i=hd[x],u;i;i=nxt[i])
{
if((u=to[i])==fa)continue;
dis[u]=dis[x]+w[i]; dfs(u,x,r);
}
}
bool dfs2(int x)
{
bool fl=,bj=;
if(vis[x])return ;
for(int i=hd[x],u;i;i=nxt[i])
{
if((u=to[i])==f[x][])continue; bj=;
bool k=dfs2(u);
if(!k)
{
fl=;
if(x==)son[++cnts].id=u,son[cnts].tim=dis[u];
}
}
if(bj)return ;
return fl;
}
bool ck(int mid)
{
memset(vis,,sizeof vis);
memset(use,,sizeof use);
memset(mn,,sizeof mn);
cnta=; cnts=;
for(int i=;i<=m;i++)
{
int x=st[i]; ll ret=mid;
for(int j=;j>=;j--)
if(f[x][j]>&&dis[x]-dis[f[x][j]]<=ret)
ret-=dis[x]-dis[f[x][j]],x=f[x][j];
if(f[x][]>||(f[x][]==&&ret<dis[x]))vis[x]=;
else
{
a[++cnta].id=i; a[cnta].tim=ret-dis[x];//-dis[x]
if(!mn[x]||a[cnta].tim<a[mn[x]].tim)mn[x]=i;//mn[1~n]=(1~m)
}
}
if(dfs2())return ;
sort(a+,a+cnta+); sort(son+,son+cnts+);
for(int i=,p=;i<=cnts;i++)
{
int nw=son[i].id;
if(mn[nw]&&!use[mn[nw]]){use[mn[nw]]=; continue;}//优先用小的,不用考虑其它儿子是因为从大往小用
while(p<=cnta&&use[a[p].id])p++;
if(p>cnta||a[p].tim<son[i].tim)return ;
use[a[p].id]=;
}
return ;
}
int main()
{
n=rd();
for(int i=,x,y,z;i<n;i++)
{
x=rd(); y=rd(); z=rd();
add(x,y,z); add(y,x,z);
sum+=z;
}
m=rd();
for(int i=;i<=m;i++)st[i]=rd();
dfs(,,);
int l=,r=sum; ans=-;
while(l<=r)
{
int mid=((l+r)>>1ll);
if(ck(mid))ans=mid,r=mid-;
else l=mid+;
}
printf("%d\n",ans);
return ;
}
