[USACO18OPEN]Disruption【洛谷P4374】

题目描述

Farmer John自豪于他所经营的交通发达的的农场。这个农场是由NNN块牧场（2≤N≤50,0002 \leq N \leq 50,0002≤N≤50,000）组成的，N−1N-1N−1条双向道路将它们连接起来，每一条道路的都为一单位长度。Farmer John注意到，从任何一块牧场到另一块牧场，都能通过一组合适的道路到达。

尽管FJ的农场现在是连通的，他担心如果有一条道路被阻断会发生什么，因为这事实上会将他的农场分为两个不相交的牧场集合，奶牛们只能够在每一个集合内移动但不能在集合间移动。于是FJ又建造了MMM条额外的双向道路（1≤M≤50,0001 \leq M \leq 50,0001≤M≤50,000），每一条的长度都是一个至多为10910^9109的正整数。奶牛们仍然可以使用原有的道路进行移动，除非其中的某些被阻断了。

如果某条原有的道路被阻断了，农场就会被分为两块不相交的区域，那么FJ就会从他的额外修建的道路中选择一条能够重建这两块区域的连通性的，取代原来那条，从而奶牛们又可以从任何一块牧场去往另一块牧场。

对于农场上每一条原有的道路，帮助FJ选出最短的替代用的道路。
输入输出格式
输入格式：

输入的第一行包含NNN和MMM。接下来的N−1N-1N−1行，每行用整数ppp和qqq描述了一条原有的道路，其中p≠qp \neq qp≠q是这条道路连接的两块牧场（在1…N1 \ldots N1…N范围内）。剩下的MMM行，每行用三个整数ppp、qqq和rrr描述了一条额外的道路，其中rrr是这条道路的长度。任何两块牧场之间至多只有一条道路。

输出格式：

对原有的N−1N-1N−1条道路的每一条，按照它们在输入中出现的顺序，输出如果这条道路被阻断的话，能够重新连接农场的最短的替代用道路的长度。如果不存在合适的替代用的道路，输出-1。

输入输出样例
输入样例#1： 复制

6 3
1 2
1 3
4 1
4 5
6 5
2 3 7
3 6 8
6 4 5

输出样例#1： 复制

7
7
8
5
5

说明

供题：Brian Dean

思路

树上的两点的简单路径上的统计(最小值)

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
const long long inf=1<<30;
const int N=50005;
inline char gc()
{
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,h[N],num=1,fa[N],dep[N],pos[N];
int dfn=0,top[N],sz[N],son[N],id[N];
struct edge
{
    int to,next;
}data[N<<1];
struct node
{
    int mn;
}tr[N<<2];
void dfs1(int x)
{
    sz[x]=1;son[x]=0;
    for(int i=h[x];i;i=data[i].next)
    {
        int y=data[i].to;
        if(y==fa[x]) 
            continue;
        fa[y]=x;dep[y]=dep[x]+1;
        id[i>>1]=y;dfs1(y);
        sz[x]+=sz[y];
        if(sz[y]>sz[son[x]]) son[x]=y;
    }
}
void dfs2(int x,int tp)
{
    pos[x]=++dfn;top[x]=tp;
    if(son[x]) dfs2(son[x],tp);
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]||y==son[x]) continue;dfs2(y,y);
    }
}
inline void build(int p,int l,int r)
{
    tr[p].mn=inf;
	if(l==r) return;int mid=l+r>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
}
inline void pushdown(int p)
{
    if(tr[p].mn==inf) return;
    tr[p<<1].mn=min(tr[p<<1].mn,tr[p].mn);
    tr[p<<1|1].mn=min(tr[p<<1|1].mn,tr[p].mn);
    tr[p].mn=inf;
}
inline int ask(int p,int l,int r,int x)
{
    if(l==r) return tr[p].mn;
    int mid=l+r>>1;
    pushdown(p);
    if(x<=mid) return ask(p<<1,l,mid,x);
    else return ask(p<<1|1,mid+1,r,x);
}
inline void cov(int p,int l,int r,int x,int y,int val)
{
    if(x<=l&&r<=y)
    {
        tr[p].mn=min(tr[p].mn,val);
        return;
    }
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) cov(p<<1,l,mid,x,y,val);
    if(y>mid) cov(p<<1|1,mid+1,r,x,y,val);
}
inline void docov(int x,int y,int val){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        cov(1,1,n,pos[top[x]],pos[x],val);
        x=fa[top[x]];
    }
    if(pos[x]>pos[y]) swap(x,y);
    if(pos[x]==pos[y]) return;
    cov(1,1,n,pos[x]+1,pos[y],val);
}
int main()
{   
    int i,x,y;
    n=read();m=read();
    for(i=1;i<n;i++)
    {
        x=read();y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }
    dfs1(1);
    dfs2(1,1);
    build(1,1,n);
    while(m--)
    {
        x=read();
        y=read();
        docov(x,y,read());
    }
    for(i=1;i<n;++i)
    {
        x=ask(1,1,n,pos[id[i]]);
        if(x==inf) 
            printf("-1\n");
        else 
            printf("%d\n",x);
    }
    return 0;
}

/*#include
#include

using namespace std;

const int maxn = 50010;
const int inf = 0x3f3f3f3f;

int n, idn[maxn], ids[maxn];

inline int min(int x, int y) {
return x < y ? x : y;
}

inline void swap(int &x, int &y) {
int t = x; x = y; y = t;
}

#define gc() (p1p2&&(p2=(p1=buf)+fread(buf,1,2000000,stdin), p1p2)?EOF:*p1++)
char buf[2000000], *p1 = buf, *p2 = buf;
inline void read(int &x) {
x = 0;
char ch = gc();
while(ch < ‘0’ || ch > ‘9’) ch = gc();
while(ch >= ‘0’ && ch <= ‘9’) x = (x<<3)+(x<<1)+(ch&15), ch = gc();
}

struct edge{
int to, nex, id;
}e[maxn<<1];
int head[maxn], tot = 0;
inline void addedge(int u, int v, int _id) {
e[tot] = (edge){v, head[u], _id}; head[u] = tot++;
e[tot] = (edge){u, head[v], _id}; head[v] = tot++;
}

int fa[maxn], d[maxn], son[maxn], size[maxn], top[maxn], dfn[maxn], rk[maxn], dfs_clock = 1;
void dfs1(int pre, int cur) {
fa[cur] = pre; d[cur] = d[pre]+1; size[cur] = 1;
for(int i = head[cur]; i != -1; i = e[i].nex)
if(e[i].to != pre) {
dfs1(cur, e[i].to);
size[cur] += size[e[i].to];
if(size[e[i].to] > size[son[cur]]) son[cur] = e[i].to;
idn[e[i].to] = e[i].id;
}
}

void dfs2(int tp, int cur) {
rk[dfs_clock] = cur; top[cur] = tp; dfn[cur] = dfs_clock++;
if(!son[cur]) return;
dfs2(tp, son[cur]);
for(int i = head[cur]; i != -1; i = e[i].nex)
if(!dfn[e[i].to])
dfs2(e[i].to, e[i].to);
}

struct seg{
int l, r, dat, laz;
}t[maxn<<4];

inline void build(int p, int l, int r) {
t[p].l = l; t[p].r = r; t[p].dat = t[p].laz = inf;
if(l == r) return;
int mid = (l+r) >> 1;
build(p<<1, l, mid);
build(p<<1|1, mid+1, r);
}

inline void pushdown(int p) {
t[p<<1].dat = min(t[p<<1].dat, t[p].laz);
t[p<<1|1].dat = min(t[p<<1|1].dat, t[p].laz);
t[p<<1].laz = min(t[p<<1].laz, t[p].laz);
t[p<<1|1].laz = min(t[p<<1|1].laz, t[p].laz);
t[p].laz = inf;
}

inline void change(int p, int l, int r, int v) {
if(l <= t[p].l && t[p].r <= r) {
t[p].dat = min(t[p].dat, v);
t[p].laz = min(t[p].laz, v);
return;
}
if(t[p].laz != inf) pushdown§;
int mid = (t[p].l + t[p].r) >> 1;
if(l <= mid) change(p<<1, l, r, v);
if(r > mid) change(p<<1|1, l, r, v);
t[p].dat = min(t[p<<1].dat, t[p<<1|1].dat);
}

inline int query(int p, int l, int r) {
if(l <= t[p].l && t[p].r <= r)
return t[p].dat;
if(t[p].laz != inf) pushdown§;
int mid = (t[p].l + t[p].r) >> 1, res = inf;
if(l <= mid) res = min(res, query(p<<1, l, r));
if(r > mid) res = min(res, query(p<<1|1, l, r));
return res;
}

inline void cg(int x, int y, int v) {
while(top[x] != top[y]) {
if(d[top[x]] < d[top[y]]) swap(x, y);
change(1, dfn[top[x]], dfn[x], v);
x = fa[top[x]];
}
if(d[x] < d[y]) swap(x, y);
change(1, dfn[y]+1, dfn[x], v);
}

int main() {
int m, u, v, w;
read(n); read(m);
memset(head, -1, sizeof(head));
for(int i = 1; i < n; i++) {
read(u); read(v);
addedge(u, v, i);
}
dfs1(0, 1);
dfs2(1, 1);
build(1, 1, n);
while(m–) {
read(u); read(v); read(w);
if(u == v) continue;
cg(u, v, w);
}
for(int i = 2; i <= n; i++) ids[idn[i]] = i;
for(int i = 1; i < n; i++) {
int t = query(1, dfn[ids[i]], dfn[ids[i]]);
printf("%d\n", t==inf ? -1 : t);
}
return 0;
}
*/