CCF-CSP认证 线性分类器 C++满分实现

代码

#include<iostream>
#include<algorithm>
using namespace std;

int main()
{
	int n,m;
	cin>>n>>m;
	char type[n+1];
	int x[n+1],y[n+1];
	int sum1=0,sum2=0;
	for(int i=1;i<=n;i++)
	{
		cin>>x[i]>>y[i]>>type[i];
		if(type[i]=='A')
		{
			sum1++;
		}
		else
		{
			sum2++;
		}
	}
	int r0,r1,r2;
	bool flag=true;
	int sum3=0,sum4=0;
	for(int i=1;i<=m;i++)
	{
		cin>>r0>>r1>>r2;
		sum3=0,sum4=0;
		for(int j=1;j<=n;j++)
		{
			if(type[j]=='A')
			{
				if(r0+r1*x[j]+r2*y[j]<0)
				{
					sum3++;
				}
				else
				{
					sum3--;
				}
			}
			else
			{
				if(r0+r1*x[j]+r2*y[j]<0)
				{
					sum4++;
				}
				else
				{
					sum4--;
				}
			}
		}
		if((abs(sum3)==sum1)&&(abs(sum4)==sum2))
		{
			cout<<"Yes";
			cout<<(i<m?"\n":"");
		}
		else
		{
			cout<<"No";
			cout<<(i<m?"\n":"");
		}
	}
	return 0;
}