牛客网 寻找三角形

[编程题] 寻找三角形

时间限制：1秒

空间限制：32768K

三维空间中有N个点，每个点可能是三种颜色的其中之一，三种颜色分别是红绿蓝，分别用'R', 'G', 'B'表示。 
现在要找出三个点，并组成一个三角形，使得这个三角形的面积最大。
但是三角形必须满足：三个点的颜色要么全部相同，要么全部不同。 

输入描述:

首先输入一个正整数N三维坐标系内的点的个数.(N <= 50) 

接下来N行，每一行输入 c x y z，c为'R', 'G', 'B' 的其中一个。x，y，z是该点的坐标。(坐标均是0到999之间的整数)

输出描述:

输出一个数表示最大的三角形面积，保留5位小数。

输入例子:

5
R 0 0 0
R 0 4 0
R 0 0 3
G 92 14 7
G 12 16 8

输出例子:

6.00000

水题，用海伦公式来求三角形面积。用枚举列出所有的情况

代码如下：

#include <stdio.h>
#include <iostream>
#include <math.h>
#include <algorithm>
using namespace std;

struct node
{
    char str;
    int x;
    int y;
    int z;
}P[50];
int main()
{
    double a,b,c,f,p;
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=0; i<n; i++) scanf(" %c %d %d %d",&P[i].str,&P[i].x,&P[i].y,&P[i].z);
        f=0.0;
        for(int i=0; i<n; i++)
            for(int j=i+1; j<n; j++)
                for(int k=j+1; k<n; k++)
                    if(P[i].str==P[j].str&&P[i].str==P[k].str)
                    {
                        a=sqrt((P[i].x-P[j].x)*(P[i].x-P[j].x)+(P[i].y-P[j].y)*(P[i].y-P[j].y)+(P[i].z-P[j].z)*(P[i].z-P[j].z));
                        b=sqrt((P[i].x-P[k].x)*(P[i].x-P[k].x)+(P[i].y-P[k].y)*(P[i].y-P[k].y)+(P[i].z-P[k].z)*(P[i].z-P[k].z));
                        c=sqrt((P[k].x-P[j].x)*(P[k].x-P[j].x)+(P[k].y-P[j].y)*(P[k].y-P[j].y)+(P[k].z-P[j].z)*(P[k].z-P[j].z));
                        p=(a+b+c)/2;
                        f=max(f,sqrt(p*(p-a)*(p-b)*(p-c)));
                    }
                    else if(P[i].str!=P[j].str&&P[i].str!=P[k].str&&P[j].str!=P[k].str)
                    {
                        a=sqrt((P[i].x-P[j].x)*(P[i].x-P[j].x)+(P[i].y-P[j].y)*(P[i].y-P[j].y)+(P[i].z-P[j].z)*(P[i].z-P[j].z));
                        b=sqrt((P[i].x-P[k].x)*(P[i].x-P[k].x)+(P[i].y-P[k].y)*(P[i].y-P[k].y)+(P[i].z-P[k].z)*(P[i].z-P[k].z));
                        c=sqrt((P[k].x-P[j].x)*(P[k].x-P[j].x)+(P[k].y-P[j].y)*(P[k].y-P[j].y)+(P[k].z-P[j].z)*(P[k].z-P[j].z));
                        p=(a+b+c)/2;
                        f=max(f,sqrt(p*(p-a)*(p-b)*(p-c)));
                    }
        printf("%.5f\n",f);
    }
    return 0;
}