BZOJ5100[POI2018]Plan metra 构造

来源：互联网发布：java实用教程编辑：程序博客网时间：2024/06/05 21:54

首先可以发现在1到n的链上的点的d1+d2是最小的这样我们可以求出1到n的链
对于其他的点他一定与这条链的上点链接可以发现对于点i和1到n上的点j 有d1[i]−d2[i]=d1[j]−d2[j] 因此可以求出每个点连向谁如果没有1到n上符合差值相等的点就不存在解

我一开始用map+sort 无线TLE+RE

#include<bits/stdc++.h>#define bug(x) cout<<(#x)<<" "<<(x)<<endlusing namespace std;const int N=5e5+5;inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}int n,cnt,tot,len=INT_MAX,mp[20000007];struct Edge{    int u,v,c;}e[N];struct node{    int s,t,id,sum;}a[N],tmp[N];vector<node>b[N];bool cmp2(node a,node b){    return a.s<b.s;}int main(){#ifdef Devil_Gary    freopen("in.txt","r",stdin);#endif    n=read();    if(n==2){        return printf("TAK\n1 2 1"),0;    }    for(int i=1;i<=n-2;i++) a[i].s=read(),a[i].id=i+1;    for(int i=1;i<=n-2;i++) a[i].t=read(),a[i].sum=a[i].s+a[i].t,len=min(len,a[i].sum);    int zjq=abs(a[1].s-a[1].t),flag=0;    if(zjq){        for(int i=2;i<=n-2;i++) if(abs(a[i].t-a[i].s)!=zjq) flag=1;        if(!flag) {            printf("TAK\n1 %d %d\n",n,zjq);            for(int i=1;i<=n-2;i++){                if(a[i].s<a[i].t)  printf("1 %d %d\n",a[i].id,a[i].s);                else printf("%d %d %d\n",n,a[i].id,a[i].t);            }             return 0;        }    }     for(int i=1;i<=n-2;i++) if(a[i].sum==len) tmp[++cnt]=a[i];    sort(tmp+1,tmp+cnt+1,cmp2);    tmp[0].id=1,tmp[0].t=tmp[0].sum=len;    tmp[++cnt].id=n,tmp[cnt].s=tmp[cnt].sum=len;    for(int i=1;i<=cnt;i++) if(tmp[i].s==tmp[i-1].s) {        return printf("NIE\n"),0;    }    for(int i=1;i<=cnt;i++){        e[++tot].u=tmp[i-1].id,e[tot].v=tmp[i].id,e[tot].c=tmp[i].s-tmp[i-1].s;    }    for(int i=0;i<=cnt;i++) mp[tmp[i].s]=i+1;    for(int i=1;i<=n-2;i++){        if(a[i].sum==len) continue;        int r=a[i].s,c=r+a[i].t-len;        r-=c/2;        if(c&1) {            return printf("NIE\n"),0;        }        else if(!mp[r])  {            return printf("NIE\n"),0;        }        else {            e[++tot].u=tmp[mp[r]-1].id,e[tot].v=a[i].id,e[tot].c=c/2;        }    }    printf("TAK\n");    for(int i=1;i<=tot;i++) printf("%d %d %d\n",e[i].u,e[i].v,e[i].c);    return 0;}

阅读全文

0 0