UVA 1325 Hypertransmission(枚举)

从小到大枚举，每枚举一条边就修改两端，那样的话就不用每次都用一个for循环去更新，时间复杂度O(n)。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=1000+10;
struct Point{
    double x,y,z;
    int sign,same,diff;
}p[maxn];
struct Dist{
    int l,r;
    double len;
}d[maxn*maxn];
int n,c;
int cmp(const Dist &a,const Dist &b) {return a.len<b.len;}
double dist(double x,double y,double z) {return sqrt(x*x+y*y+z*z);}
void intial()
{
    c=0;
    for(int i=0;i<n;i++)
    {
        scanf("%lf%lf%lf%d",&p[i].x,&p[i].y,&p[i].z,&p[i].sign);
        p[i].same=1;p[i].diff=0;
        for(int j=0;j<i;j++)
        {
            d[c].l=i;d[c].r=j;
            d[c++].len=dist(p[i].x-p[j].x,p[i].y-p[j].y,p[i].z-p[j].z);
        }
    }
    sort(d,d+c,cmp);
}
void solve()
{
    int ans=0,tmp=0;
    double R=0;
    for(int i=0;i<c;i++)
    {
        int t=d[i].l,k=d[i].r;
        if(p[t].sign==p[k].sign){
            if(p[t].diff-p[t].same==1) tmp--;
            if(p[k].diff-p[k].same==1) tmp--;
            p[t].same++;p[k].same++;
        }
        else{
            if(p[t].same==p[t].diff) tmp++;
            if(p[k].same==p[k].diff) tmp++;
            p[t].diff++;p[k].diff++;
        }
        if(i!=c-1&&fabs(d[i].len-d[i+1].len)<1e-9) continue;//现在是每枚举一条边修改一下，但是如果下一条边一样，那么修改是不全的
        if(tmp>ans) {ans=tmp;R=d[i].len;}
    }
    printf("%d\n%.4f\n",ans,R);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        intial();
        solve();
    }
    return 0;
}