bzoj5100 [POI2018]Plan metra 构造

5100: [POI2018]Plan metra

Time Limit: 40 Sec  Memory Limit: 128 MBSec  Special Judge
Submit: 189  Solved: 43
[Submit][Status][Discuss]

Description

有一棵n个点的无根树，每条边有一个正整数权值，表示长度，定义两点距离为在树上的最短路径的长度。

已知2到n-1每个点在树上与1和n的距离，请根据这些信息还原出这棵树。

 

Input

第一行包含一个正整数n(2<=n<=500000)，表示点数。

第二行包含n-2个正整数d(1,2),d(1,3),...,d(1,n-1)，分别表示每个点到1的距离。

第三行包含n-2个正整数d(n,2),d(n,3),...,d(n,n-1)，分别表示每个点到n的距离。

输入数据保证1<=d<=1000000。

 

Output

若无解，输出NIE。

否则第一行输出TAK，接下来n-1行每行三个正整数u,v,c(1<=u,v<=n,1<=c<=1000000)

表示存在一条长度为c的连接u和v两点的树边。

若有多组解，输出任意一组。

 

Sample Input

7
6 6 2 2 1
5 3 5 1 4

Sample Output

TAK
1 5 2
5 7 1
5 2 4
7 3 3
1 4 2
1 6 1

HINT

 

Source

鸣谢Claris上传试题

构造题
分情况讨论
1. 1-n路径上没有其他点，即1与n直接相连
如果是这种情况，那么就有：对于任意一个点 i 1<i<n i到1与n的距离之差为一个定值a,a=1到n的边权
特判一下 每个点到 1和n的距离差是否相等就好
如果相等，可以直接把1和n相连 其它点都与1和n相连
如果不等，进入第二种情况

2. 1-n路径上还有点
我们可以找出这些点 这些点有一个共性：它们 到1与n的距离和 相等 且 比 不在1-n路径上的点 到1和n的距离和 小
找到这些点之后 其他点都可以通过与路径上的点相连来构造答案
如果一个点i与路径上某点相连 设它到1的距离为d1 到n的距离为dn

那么它到路径上某点（假设存在）的距离就是 w=（d1+dn-（1-n的距离））/2
/2是因为计算了2次 而如果w不能被整除的话，点i不能链接到树上

再判断路径上这个点是否存在，如果路径上不存在这个点，依旧没法把点i链接到树 判断条件 ： 到1距离为d1-w的点存在于路径上

这样就一定可以构造出一棵合法树

#include<bits/stdc++.h>
#define N 500005
using namespace std;
int n,fg,mn=0x3f3f3f3f,fa[N],w[N],d1[N],dn[N],id[N<<];

inline char gc(){
    static char s[],*p1,*p2;
    if(p1==p2)p2=(p1=s)+fread(s,,,stdin);
    if(p1==p2)return EOF;return *p1++;
}
inline int read(){
    int x=;char ch=gc();
    while(ch<''||ch>'')ch=gc();
    while(ch>=''&&ch<='')x=x*+ch-'',ch=gc();
    return x;
}
inline int solve1(){
    int dis=abs(d1[]-dn[]);
    if(!dis)return ;
    for(register int i=;i<n;++i)
    if(dis!=abs(d1[i]-dn[i]))return ;
    puts("TAK");printf("%d %d %d\n",,n,dis);
    for(register int i=;i<n;++i){
        if(d1[i]<dn[i])printf("%d %d %d\n",i,,d1[i]);
        else printf("%d %d %d\n",i,n,dn[i]);
    }
    return ;
}

int main(){
    n=read();if(n==){puts("TAK\n1 2 1");return ;}
    for(register int i=;i<n;++i)d1[i]=read();
    for(register int i=;i<n;++i){
        dn[i]=read();
        dn[i]+d1[i]<mn?mn=dn[i]+d1[i]:;
    }
    if(solve1())return ;
    id[mn]=n;id[]=;
    for(register int i=;i<n&&!fg;++i){
        if(dn[i]+d1[i]==mn){
            if(id[d1[i]])fg=;
            else id[d1[i]]=i;
        }
    }
    if(fg){puts("NIE");return ;}
    for(register int i=;i<n&&!fg;++i){
        if(d1[i]+dn[i]==mn)continue;
        int tmp=d1[i],len=d1[i]+dn[i]-mn;
        if(len&)fg=;len>>=;
        tmp-=len;if(!id[tmp])fg=;
        fa[i]=id[tmp];w[i]=len;
    }
    if(fg){puts("NIE");return ;}
    int pre=;puts("TAK");
    for(register int i=;i<=mn;++i){
        if(!id[i])continue;
        printf("%d %d %d\n",pre,id[i],i-d1[pre]);
        pre=id[i];
    }
    for(register int i=;i<n;++i){
        if(!fa[i])continue;
        printf("%d %d %d\n",i,fa[i],w[i]);
    }
    return ;
}