CSP202006-1线性分类器 满分代码

题目

满分代码

思路如下：
用结构体表示每一个点，将点带入直线，如果大于0，标志da=1；否则标志da=0。用两个数组分别存放两个类别的标志da，接下来判断两个数组里的数是否一致，如果一致，说明当前直线可以划分这两类数据，输出Yes，否则输出No。

#include<stdio.h>
struct Note   //用结构体表示点
{
	int x;
	int y;
	char type;   //当前点的类别
	int da = 0;  //标志da表示当前点带入直线，看是否大于0，若大于0，置为1，否则是0
};
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	struct Note note[n];
	for(int i=0;i<n;i++)
		scanf("%d%d %c",&note[i].x,&note[i].y,&note[i].type);
	
	for(int j=0;j<m;j++)
	{
		int a = 0,b = 0;
		int r0,r1,r2;
		bool xian = true;
		scanf("%d%d%d",&r0,&r1,&r2);
		for(int i=0;i<n;i++)  //遍历每一个点，统计两个类别的数量
		{
			
			if(note[i].type == 'A')
				a++;
			else
				b++;
			int k = r0 + r1 * note[i].x + r2 * note[i].y;
			if(k>0)
				note[i].da = 1;
			else
				note[i].da = 0;
		}
		int d[a] = {0},e[b] = {0};  //用两个数组分别存放两个类别的标志da
		int p = 0,q = 0;
		for(int i=0;i<n;i++)
		{
			
			if(note[i].type == 'A')
			{
				d[p] = note[i].da;
				p++;
			}
			else
			{
				e[q] = note[i].da;
				q++;
			}
		}
		//接下来判断两个数组里的数是否一致，如果一致，说明当前直线可以划分这两类数据，输出Yes，否则输出No
		int s = d[0];
		for(int h=0;h<a;h++)
		{
			if(d[h] == s)
				continue;
			else
			{
				xian = false;
				break;
			}
				
		}
		int l = e[0];
		for(int h=0;h<b;h++)
		{
			if(e[h] == l)
				continue;
			else
			{
				xian = false;
				break;
			}
		}
		if(xian == false)
			printf("No\n");
		else
			printf("Yes\n");
		xian = true; 
	}
	return 0;
}

例题输入：
9 3
1 1 A
1 0 A
1 -1 A
2 2 B
2 3 B
0 1 A
3 1 B
1 3 B
2 0 A

结果：