构造题,细节较多。
第一次WA:没有判1-n的链上如果d[1][i]相同的话是不合法的。
第二次WA:没有判1-n距离为0的情况是不合法的。
#include <bits/stdc++.h>
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;
}