【JZOJ4894】【NOIP2016提高A组集训第16场11.15】SJR的直线

题目描述

数据范围

解法

考虑逐次加入每一条直线。
对于当前已加入的直线集合L，现在要新加入一条直线l。
那么它产生的贡献，与平行线有关。
对于任意三条直线，如果其中任意两条平行，那么将不做贡献。
所以给原直线的斜率排序，然后O(n)逐次加入，就能快速求答案了。

代码

#include<iostream>#include<stdio.h>#include<string.h>#include<algorithm>#include<math.h>#define ll long long#define sqr(x) ((x)*(x))using namespace std;const char* fin="trokuti.in";const char* fout="trokuti.out";const int inf=0x7fffffff;const int maxn=400007,mo=1000000007;int n,i,j,k;ll ans;struct line{    ll x,y,z;    bool friend operator ==(const line &a,const line &b){        return (ll)a.x*b.y==(ll)a.y*b.x;    }}a[maxn];bool cmp(const line &a,const line &b){    /*int k=1;    if (b.y<0) k*=-1;    if (a.y<0) k*=-1;    return (ll)a.x*abs(b.y)*k<(ll)b.x*abs(a.y);*/    if (a.y==0 && b.y==0) return false;    else if (a.y==0) return false;    else if (b.y==0) return true;    return a.x*1.0/a.y<b.x*1.0/b.y;}void qsort(int l,int r){    int i=l,j=r;    line mid=a[(l+r)/2];    while (i<=j){        while (cmp(a[i],mid)) i++;        while (cmp(mid,a[j])) j--;        if (i<=j){            swap(a[i],a[j]);            i++;            j--;        }    }    if (i<r) qsort(i,r);    if (l<j) qsort(l,j);}ll count(int x){    return (ll)x*(x-1)/2;}int main(){    freopen(fin,"r",stdin);    freopen(fout,"w",stdout);    scanf("%d",&n);    for (i=1;i<=n;i++){        scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);        //a[i].x+=1000000000;        //a[i].y+=1000000000;    }    //sort(a+1,a+n+1,cmp);    //for (i=1;i<=n;i++) for (j=i+1;j<=n;j++) if (!cmp(a[i],a[j])) swap(a[i],a[j]);    qsort(1,n);    j=0;    k=0;    for (i=1;i<=n;i++){        if (i>1 && a[i]==a[i-1]) j++;        else k+=(count(j+1))%mo,j=0;        ans=(ans+count(i-1-j)-k)%mo;    }    cout<<ans<<endl;    return 0;}

启发

垃圾STL，不判断边界条件，不得不手打QSORT。