一道构造杂题
画画图可以发现,把1到n的路径提出来,其他的点都是跟这上面的其中一个点相连的。
这个路径就是把点权设为 ai−bi 后排序(除了1和n),然后相同点权中 ai 最小(或者 bi 最小) 的点一次相连。然后把1和n插到这个序列中。
显然序列中1的左边和n的右边最多只有1个点。
那么枚举1的左边有几个点,n的右边有几个点,dfs判一判就好了
O(n)
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=500010;
int n,cnt=1,a[N],b[N],id[N],G[N],ia[N],ib[N];
struct edge{
int t,nx,w;
}E[N<<1];
inline bool cmp(const int &x,const int &y){
return a[x]-b[x]<a[y]-b[y];
}
inline void addedge(int x,int y,int z){
E[++cnt].t=y; E[cnt].nx=G[x]; E[cnt].w=z; G[x]=cnt;
E[++cnt].t=x; E[cnt].nx=G[y]; E[cnt].w=z; G[y]=cnt;
}
void dfs(int x,int f,int *a){
for(int i=G[x];i;i=E[i].nx)
if(E[i].t!=f){
a[E[i].t]=a[x]+E[i].w;
dfs(E[i].t,x,a);
}
}
inline char nc(){
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &x){
char c=nc(); x=0;
for(;c>'9'||c<'0';c=nc());for(;c>='0'&&c<='9';x=x*10+c-'0',c=nc());
}
inline bool solve(int A,int B){
memset(G,0,sizeof(G)); cnt=1;
int lst=1;
for(int i=2,j;i<n;i=j){
for(j=i;a[id[i]]-b[id[i]]==a[id[j]]-b[id[j]] && j<n;j++);
if(i==2 && A) for(int k=i;k<j;k++) addedge(1,id[k],a[id[k]]);
else if(j>=n && B) for(int k=i;k<j;k++) addedge(n,id[k],b[id[k]]);
else{
int p=id[i];
for(int k=i+1;k<j;k++)
if(a[id[k]]<a[p]) p=id[k];
if(a[p]<=a[lst]) return 0;
addedge(lst,p,a[p]-a[lst]); lst=p;
for(int k=i;k<j;k++)
if(p!=id[k]){
if(a[id[k]]==a[p]) return 0;
addedge(id[k],p,a[id[k]]-a[p]);
}
}
}
if(lst!=1) addedge(lst,n,b[lst]);
else addedge(lst,n,b[2]<a[2]?a[2]-b[2]:b[2]-a[2]);
dfs(1,0,ia); dfs(n,0,ib);
for(int i=2;i<n;i++)
if(a[i]!=ia[i]) return 0;
for(int i=2;i<n;i++)
if(ib[i]!=b[i]) return 0;
puts("TAK");
for(int i=2;i<=cnt;i+=2) printf("%d %d %d\n",E[i].t,E[i^1].t,E[i].w);
return 1;
}
int main(){
freopen("1.in","r",stdin);
freopen("1.out","w",stdout);
read(n);
if(n==2) return puts("TAK\n1 2 1"),0;
for(int i=2;i<n;i++) read(a[i]);
for(int i=2;i<n;i++) read(b[i]);
for(int i=1;i<=n;i++) id[i]=i;
sort(id+2,id+n,cmp);
for(int i=0;i<=1;i++)
for(int j=0;j<=1;j++)
if(solve(i,j)) return 0;
puts("NIE");
return 0;
}