CF 851C.Five Dimensional Points 点积,思维

最新推荐文章于 2020-02-13 14:56:20 发布

原创最新推荐文章于 2020-02-13 14:56:20 发布 · 280 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Codeforces 泛做

构造 --- 规律 --- 思维 --- 贪心 ---

本文介绍了一种算法，用于确定5维空间中特定条件下被定义为good的点的数量。通过分析点之间的角度关系，使用点积判断是否构成锐角，最终找出符合条件的点。文中提供了两种实现方案，一种是O(n^3)的暴力解法，另一种是更高效的O(n^2)算法。

题意:5维空间上有n个点,若一个点a,能找到两个点b,c使得向量ab和向量ac的夹角为锐角则这个点为bad.否则为good
n<=1e3.问这n个点有多少为good.

判断两个向量夹角,用到点积 ab * ac =|ab|*|ac|*sin$ 若为锐角则点积>0
暴力枚举点a 在枚举b,c判断点积 O(n^3) 然后过了???.

看了下题解,原来n>11时答案都为0.(二维答案为1时最多5个点).
然后评论区大佬由有一种O(n^2)的方法. 因为三角形至少有两个锐角,所以最多只有两个good点

先求出这两个good点在O(n^2)判断即可

#include <bits/stdc++.h>
using namespace std;
typedef double db;
const int N=2e5+5;
struct Point{
	db a,b,c,d,e;
	Point(db va,db vb,db vc,db vd,db ve)
	{
		a=va,b=vb,c=vc,d=vd,e=ve;
	}
	Point()
	{}
	 
}p[N];
Point operator-(const Point& v1,const Point& v2)
{
	return Point(v2.a-v1.a,v2.b-v1.b,v2.c-v1.c,v2.d-v1.d,v2.e-v1.e);
}
double dot(Point v1,Point v2)
{
	return v1.a*v2.a+v1.b*v2.b+v1.c*v2.c+v1.d*v2.d+v1.e*v2.e;
}
int n;
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	vector<int> res;
	cin>>n;
	for(int i=1;i<=n;i++)
		cin>>p[i].a>>p[i].b>>p[i].c>>p[i].d>>p[i].e;
	if(n>11)
	{
		puts("0");
		return 0;
	}
	for(int i=1;i<=n;i++)
	{
		bool flag=true;
		for(int j=1;j<=n&&flag;j++)
		{
			if(i==j)	continue;	
			for(int k=1;k<=n;k++)
			{
				if(i==k||j==k)	continue;
				Point v1=p[j]-p[i],v2=p[k]-p[i];
				if(dot(v1,v2)>0)
				{
					flag=false;
					break;
				} 
			}
		}
		if(flag)
			res.push_back(i);
	}
	printf("%d\n",res.size());
	for(int i=0;i<res.size();i++)
		printf("%d ",res[i]);
	return 0;
}

大佬の代码:

#include <bits/stdc++.h>

#define MN 1001000
#define w1 while(1)
#define debug(a) cerr << a << endl
#define pb push_back
#define mp make_pair

using namespace std;
typedef long long ll;

int n, a[MN][6], bad[MN], b, c, d;
queue<int> s;
vector<int> ans;
//char s[MN]; string str;

int scalar(int i, int j, int p) {
    return (a[i][0] - a[p][0])*(a[j][0] - a[p][0])
            + (a[i][1] - a[p][1])*(a[j][1] - a[p][1])
            + (a[i][2] - a[p][2])*(a[j][2] - a[p][2])
            + (a[i][3] - a[p][3])*(a[j][3] - a[p][3])
            + (a[i][4] - a[p][4])*(a[j][4] - a[p][4]);
}

int main() {
    scanf("%d", &n);

    for(int i=0; i<n; i++) {
        for(int j=0; j<5; j++) {
            scanf("%d",&a[i][j]);
            s.push(i);
        }
    }

    while(!s.empty()) {
        if(s.size() > 2) {
            b = s.front(); s.pop();
            c = s.front(); s.pop();
            d = s.front(); s.pop();

            if(scalar(d, b, c) <= 0) {
                bad[d] = 1;
                bad[b] = 1;
                s.push(c);
            }

            else if(scalar(d, c, b) <= 0) {
                bad[d] = 1;
                bad[c] = 1;
                s.push(b);
            }

            else if(scalar(b, c, d) <= 0) {
                bad[b] = 1;
                bad[c] = 1;
                s.push(d);
            }

            else {
                bad[d] = 1;
                bad[b] = 1;
                bad[c] = 1;
            }
        }

        else {
            b = s.front(); s.pop();
            int flag = 1;

            for(int i = 0; i < n && flag; i++) {
                if(i == b) continue;
                for(int j = 0; j < n && flag; j++) {
                    if(j == b || j == i) continue;
                    if(scalar(i, j, b) > 0) flag = 0;
                }
            }

            if(flag) ans.push_back(b);
            else bad[b] = 1;
        }
    }

    sort(ans.begin(), ans.end());
    printf("%d\n", ans.size());
    for(int x : ans) {
        printf("%d\n", x + 1);
    }
    return 0;
}