题目链接
题目描述
平面上有 N N N 条直线, 其中第 i i i 条直线是 y = A i ⋅ x + B i y = A_i \cdot x + B_i y=Ai⋅x+Bi
请计算这些直线将平面分成了几个部分。
输入格式
第一行包含一个整数 N N N。
以下 N N N 行, 每行包含两个整数 A i , B i A_i,B_i Ai,Bi。
输出格式
一个整数代表答案。
输入输出样例
输入
3
1 1
2 2
3 3
输出
6
数据范围
- 1 ≤ N ≤ 1000 1 \leq N \leq 1000 1≤N≤1000
- 1 0 − 5 ≤ A i , B i ≤ 1 0 5 10^{-5} \leq A_i,B_i \leq 10^5 10−5≤Ai,Bi≤105
解法:计算几何 + 计数
如上图,先有
4
4
4 根绿色的线,然后又来了一根黑色的线,与之前绿色的线共有
4
4
4 个交点。将原来的
5
5
5 个部分,一分为二,即又新增了
5
5
5 个部分(分别是
1
,
2
,
3
,
4
,
5
1,2,3,4,5
1,2,3,4,5)。
我们可以得出一个结论:如果新来的一条直线 与 已有的直线有 t t t 个交点,那么就会增加 t + 1 t + 1 t+1 个部分。
我们直接枚举每一条直线,与前面直线的交点即可。
注意:可能会出现重边的情况,所以需要去重!
时间复杂度: O ( n 2 ) O(n^2) O(n2)
C++代码:
#include <iostream>
#include <set>
#include <vector>
using namespace std;
using LD = long double;
using PII = pair<int,int>;
int main(){
int n;
cin>>n;
vector<LD> k, b;
set<PII> s;
for(int i = 0;i < n;i++){
int kk,bb;
cin>>kk>>bb;
s.insert(make_pair(kk,bb));
}
for(auto& [kk,bb]:s){
k.push_back(kk);
b.push_back(bb);
}
n = k.size();
set<pair<LD,LD>> st;
int ans = 2;
for(int i = 1;i < n;i++){
for(int j = i - 1;j >= 0;j--){
if(k[i] == k[j]) continue; //两条边平行,肯定没有交点
LD x = (b[i] - b[j]) / (k[j] - k[i]);
LD y = k[i] * x + b[i];
st.insert(make_pair(x, y));
}
ans += st.size() + 1;
st.clear();
}
cout<<ans<<'\n';
return 0;
}
Java代码:
import java.util.*;
import java.io.*;
class Pair {
public double x;
public double y;
public Pair(double _x, double _y) {
this.x = _x;
this.y = _y;
}
@Override
public boolean equals(Object obj) {
if (this == obj) return true;
if (obj == null || getClass() != obj.getClass()) return false;
Pair p = (Pair)obj;
return (p.x == x) && (p.y == y);
}
@Override
public int hashCode() {
return Objects.hash(x, y);
}
}
public class Main {
static BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
public static void main(String[] args) throws Exception
{
int n = Integer.parseInt(in.readLine().trim());
double[] k = new double[n];
double[] b = new double[n];
String[] str;
for(int i = 0;i < n;i++) {
str = in.readLine().split(" ");
k[i] = Double.parseDouble(str[0]);
b[i] = Double.parseDouble(str[1]);
for(int j = i - 1;j >= 0;j--) {
if(k[i] == k[j] && b[i] == b[j]) {
n--;
i--;
break;
}
}
}
Set<Pair> st = new HashSet<>();
int ans = 2;
for(int i = 1;i < n;i++) {
for(int j = i - 1;j >= 0;j--) {
if(k[i] == k[j]) continue;
double x = (b[i] - b[j]) / (k[j] - k[i]);
double y = k[i] * x + b[i];
st.add(new Pair(x, y));
}
ans += st.size() + 1;
st.clear();
}
System.out.println(ans);
}
}