[构造] BZOJ5100 [POI2018]. Plan metra

一道构造杂题

画画图可以发现，把1到n的路径提出来，其他的点都是跟这上面的其中一个点相连的。

这个路径就是把点权设为 $a_i-b_i$ 后排序（除了1和n），然后相同点权中 $a_i$ 最小（或者 $b_i$ 最小） 的点一次相连。然后把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;
}