【CCF CSP历年考试带注释全题解 C++版】202006-1 线性分类器

#include 
#include 
using namespace std;

int sign(pair point, vector para) { //方便直接输出函数正负号
	int value = para[0] + point.first * para[1] + point.second * para[2];
	return value > 0 ? 1 : (value == 0 ? 0 : -1); //1代表正，-1代表负，0代表0
}

int main() {
	int n, m;
	cin >> n >> m; //输入n，m
	vector> type_A; //这里直接根据type不同分为两类，用pair对付坐标点再适合不过
	vector> type_B;
	for (int i = 0; i < n; i++) { //输入n组点信息
		int x, y;
		char type;
		cin >> x >> y >> type;
		if (type == 'A') type_A.push_back(pair(x, y));
		else type_B.push_back(pair(x, y));
	}
	vector> parameter(m, vector(3));
	for (int i = 0; i < m; i++) { //输入m组参数信息
		cin >> parameter[i][0] >> parameter[i][1] >> parameter[i][2];
	}
	//现在有个问题，图像上A、B点谁上谁下没有确定，因此根据第一个A点来判断，若A点为空集则根据第一个B点判断，
	//然后保证A点全在一侧，B点全在另一侧，数值上来看保证将A、B坐标代入方程后值符号相反且不含有0，
	//特别注意空集的情况
	for (vector para : parameter) {
		int sign_of_A = 0;
		if (!type_A.empty())  sign_of_A = sign(type_A[0], para);
		else sign_of_A = -sign(type_B[0], para);
		if (sign_of_A == 0) { //表示第一个A点或第一个B点值为0，直接输出No
			cout << "No" << endl;
			continue;
		}
		int count = 0; //用于统计符合条件的点数
		for (pair p : type_A) { //统计符号相同的A点数
			if (sign(p, para) == sign_of_A) count++;
		}
		for (pair p : type_B) { //统计符号相同的B点数
			if (sign(p, para) == -sign_of_A) count++;
		}
		cout << (count == n ? "Yes" : "No") << endl; //将符合条件的点数和总点数对比，若一样则输出Yes
	}
}