Codeforces Round #558 (Div. 2) C2. Power Transmission (Hard Edition)

链接

https://codeforces.com/contest/1163/problem/C2

题解

如果我找出了所有的直线，比方说有$m$条
那么答案就是用$C_m^2$减去所有平行的选法

问题是怎么求出直线的条数

枚举点$i$，然后枚举直线上另一个点$j$，对于每种斜率的直线，我都记录这种直线上$j$的最小值，如果发现最小值大于$i$，就说明$i$是这条直线上编号最小的点，此时把这条直线的斜率加入总的$map$里

代码

#include <bits/stdc++.h>
using namespace std;
#define maxn 1000010
#define eps 1e-8
typedef long long ll;
typedef pair<ll,ll> pll;
ll x[maxn], y[maxn];
map<pll,ll> cnt, tmp;
int main()
{
    ll n, i, j, a, b, l=0;
    cin>>n;
    for(i=1;i<=n;i++)cin>>x[i]>>y[i];
    for(i=1;i<=n;i++)
    {
        tmp.clear();
        for(j=1;j<=n;j++)
        {
            if(i==j)continue;
            ll x1, y1, x2, y2;
            if(x[i]<x[j])
            {
                x1=x[i], x2=x[j];
                y1=y[i], y2=y[j];
            }
            else if(x[i]>x[j])
            {
                x1=x[j], x2=x[i];
                y1=y[j], y2=y[i];
            }
            else
            {
                x1=x[i], x2=x[j];
                y1=min(y[i],y[j]), y2=max(y[i],y[j]);
            }
            
            ll dx=x2-x1, dy=y2-y1, g=__gcd(dx,abs(dy));
            dx/=g;
            dy/=g;
            if(tmp.find(pll(dx,dy))==tmp.end() or tmp[pll(dx,dy)]>j)
                tmp[pll(dx,dy)]=j;
        }
        for(auto pr:tmp)
            if(pr.second>i)
                cnt[pr.first]++, l++;
    }
    ll ans=l*(l-1)/2;
    for(auto pr:cnt)
    {
        ll c=pr.second;
        ans-=c*(c-1)/2;
    }
    cout<<ans<<endl;
    return 0;
}