BZOJ4246: 两个人的星座

Description

JOI酱和IOI酱是好朋友。某天，JOI酱与IOI酱决定去山上的某个展望台进行天体观测。

从展望台上可以观测到N颗星星，编号为1...N。每颗星星的颜色为红色、蓝色、黄色中的一种。

在展望台上观测到的星星可以用坐标系上的点来表示。在坐标系上，星i(1<=i<=N)对应的点为Pi(Xi,Yi)。坐标系上的点两两不同，且不存在三点共线。

JOI酱和IOI酱想要设立一个叫做“JOIOI座”的星座。首先。两个人决定使用红色、蓝色、黄色三种颜色的星各一个构成的三角形。他们将这样的三角形称作“好三角形”。

两人将满足以下条件的好三角形无序二元组作为JOIOI座的候补：

两个三角形没有公共点(包括内部和边界)。换言之，两个三角形之间既不相交，也不存在某个三角形包含另一个三角形。

 

JOI酱和IOI酱想知道构成JOIOI座的候补一共有多少种方案。

注意如果构成三角形的6个点一样但是构成三角形的方式不同，算作不同的方案。

现在给出展望台上能观测到的星星的信息，请求出构成JOIOI座的候补一共有多少种方案

Input

第一行一个整数N，代表展望台上能观测到的星星的数量。

接下来N行，第i行(1<=i<=N)有三个空格分隔的整数Xi,Yi,Ci，表示星i的坐标为Pi(Xi,Yi)，Ci表示星i的颜色，其中0代表红色，1代表蓝色，2代表黄色。

Output

输出一行一个整数，表示JOIOI座候补的方案数。

Sample Input

7
0 0 0
2 0 1
1 2 2
-2 1 0
-2 -3 0
0 -2 1
2 -2 2

Sample Output

4

HINT

样例中，JOIOI的候补有以下四种方案：

 

6<=N<=3000

-10^5<=Xi<=10^5(1<=i<=N)

-10^5<=Yi<=10^5(1<=i<=N)

0<=Ci<=2(1<=i<=N)

每种颜色的星至少存在一个

Pi≠Pj(1<=i<j<=N)

Pi,Pj,Pk不共线(1<=i<j<k<=N)

请注意你的常数

Source

计算几何

考虑两个三角形不相交

那么它们一定可以选出两对点连成的线把它们分割到两个半平面

枚举一个点，另一边按极角序排序，统计答案

最后答案要除以4，因为每对三角形被算了4次

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 3030;
const double pi = acos( -1 );

struct Point
{
	int x, y, c, id;
	double k;
	bool operator < ( const Point &b ) const { return k < b.k; }
}p[MAXN], o;

int n, cnt[2][3], bl[MAXN];

long long ans, tmp;

int main()
{
	scanf( "%d", &n );
	for( int i = 1 ; i <= n ; i++ ) scanf( "%d%d%d", &p[ i ].x, &p[ i ].y, &p[ i ].c ), p[ i ].id = i;
	for( int i = 1 ; i <= n ; i++ )
	{
		for( int j = 1 ; j <= n ; j++ ) if( p[ j ].id == i ) { o = p[ j ]; break; }
		int col = o.c;
		for( int j = 1 ; j <= n ; j++ )
		{
			p[ j ].k = ( p[ j ].id == i ) ? 1e9 : atan2( p[ j ].y - o.y, p[ j ].x - o.x );
			if( p[ j ].k <= 0 ) p[ j ].k += pi;
		}
		memset( cnt, 0, sizeof cnt );
		sort( p + 1, p + n + 1 );
		for( int j = 1 ; j < n ; j++ )
			if( p[ j ].y < o.y || ( p[ j ].y == o.y && p[ j ].x > o.x ) ) cnt[ bl[ j ] = 0 ][ p[ j ].c ]++;
			else cnt[ bl[ j ] = 1 ][ p[ j ].c ]++;
		for( int j = 1 ; j < n ; j++ )
		{
			cnt[ bl[ j ] ][ p[ j ].c ]--;
			tmp = 1;
			if( col ) tmp *= cnt[ 0 ][ 0 ]; if( p[ j ].c ) tmp *= cnt[ 1 ][ 0 ];
			if( col ^ 1 ) tmp *= cnt[ 0 ][ 1 ]; if( p[ j ].c ^ 1 ) tmp *= cnt[ 1 ][ 1 ];
			if( col ^ 2 ) tmp *= cnt[ 0 ][ 2 ]; if( p[ j ].c ^ 2 ) tmp *= cnt[ 1 ][ 2 ];
			ans += tmp;
			tmp = 1;
			if( col ) tmp *= cnt[ 1 ][ 0 ]; if( p[ j ].c ) tmp *= cnt[ 0 ][ 0 ];
			if( col ^ 1 ) tmp *= cnt[ 1 ][ 1 ]; if( p[ j ].c ^ 1 ) tmp *= cnt[ 0 ][ 1 ];
			if( col ^ 2 ) tmp *= cnt[ 1 ][ 2 ]; if( p[ j ].c ^ 2 ) tmp *= cnt[ 0 ][ 2 ];
			ans += tmp;
			cnt[ bl[ j ] ^= 1 ][ p[ j ].c ]++; 
		}
	}
	cout << ans / 4 << endl;
}