bzoj5100 [POI2018]Plan metra（树+构造）

构造题，细节较多。
第一次WA：没有判1-n的链上如果d[1][i]相同的话是不合法的。
第二次WA：没有判1-n距离为0的情况是不合法的。

#include 
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 500010
int n,d[2][N],mn=inf,ansid[N],ansd[N];
vector<int>a;
vector<int>::iterator it;
inline int abs(int x){return x<0?-x:x;}
inline bool cmp(int x,int y){return d[0][x]<d[0][y];}
inline bool cmp1(int x,int val){return d[0][x]<val;}
int main(){
//	freopen("a.in","r",stdin);
	scanf("%d",&n);
	if(n==2){puts("TAK");printf("%d %d %d\n",1,2,1);return 0;}
	for(int i=2;i<=n-1;++i) scanf("%d",&d[0][i]);
	for(int i=2;i<=n-1;++i) scanf("%d",&d[1][i]);
	int tmp=abs(d[0][2]-d[1][2]);bool flag=1;
	for(int i=2;i<=n-1;++i){
		mn=min(mn,d[0][i]+d[1][i]);
		if(abs(d[0][i]-d[1][i])!=tmp) flag=0;
	}
	if(flag&&tmp!=0){
		puts("TAK");
		printf("%d %d %d\n",1,n,tmp);
		for(int i=2;i<=n-1;++i){
			if(d[0][i]<d[1][i]) printf("%d %d %d\n",i,1,d[0][i]);
			else printf("%d %d %d\n",i,n,d[1][i]);
		}return 0;
	}
	d[0][n]=mn;d[0][1]=0;d[1][n]=0;d[1][1]=mn;
	for(int i=1;i<=n;++i)
		if(mn==d[0][i]+d[1][i]) a.push_back(i);
	sort(a.begin(),a.end(),cmp);
	for(int i=0;i+1<a.size();++i)
		if(d[0][a[i+1]]-d[0][a[i]]==0){puts("NIE");return 0;}
	for(int i=2;i<=n-1;++i){
		int d1=d[0][i]+d[1][i]-mn;
		if(!d1) continue;
		if(d1&1){puts("NIE");return 0;}
		d1>>=1;int d2=d[0][i]-d1;
		it=lower_bound(a.begin(),a.end(),d2,cmp1);
		if(it==a.end()||d[0][*it]!=d2){puts("NIE");return 0;}
		ansid[i]=*it;ansd[i]=d1;
	}puts("TAK");
	for(int i=0;i+1<a.size();++i) printf("%d %d %d\n",a[i],a[i+1],d[0][a[i+1]]-d[0][a[i]]);
	for(int i=2;i<=n-1;++i){
		if(d[0][i]+d[1][i]==mn) continue;
		printf("%d %d %d\n",i,ansid[i],ansd[i]);
	}
	return 0;
}