@4-1 CCF 2020-06-1 线性分类器

题目描述

样例

源代码

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

vector<int> A1, A2;
vector<int> B1, B2;

int main()
{
    int N, M;
    cin >> N >> M;
    while (N--)
    {
        int x, y;
        string C;
        cin >> x >> y;
        cin >> C;
        if (C == "A")
        {
            A1.push_back(x);  //在A1的第一个元素（从第0个算起）位置插入数值5
            A2.push_back(y);
        }
        else
        {
            B1.push_back(x);
            B2.push_back(y);
        }
    }
    while (M--)
    {
        bool judge = true;
        int a, b, c;
        cin >> a >> b >> c;
        int AA = 0, BB = 0;
        if (a + A1[0] * b + A2[0] * c > 0) AA = 1;     //这里先判断第一个数 确定其在直线上方还是下方
        else AA = -1;
        for (int i = 1; i < A1.size(); i++)  //A1.size() 返回A1在内存中总共可以容纳的元素个数  
        {
            if (a + A1[i] * b + A2[i] * c < 0 && AA == 1) judge = false;
            if (a + A1[i] * b + A2[i] * c > 0 && AA == -1) judge = false;
        }

        if (a + B1[0] * b + B2[0] * c > 0) BB = 1;
        else BB = -1;
        for (int i = 1; i < B1.size(); i++)
        {
            if (a + B1[i] * b + B2[i] * c < 0 && BB == 1) judge = false;
            if (a + B1[i] * b + B2[i] * c > 0 && BB == -1) judge = false;
        }
        if (judge)
        {
            if (AA * BB < 0)
            {
                cout << "Yes" << endl;
            }
            else
            {
                cout << "No" << endl;
            }
        }
        else
        {
            cout << "No" << endl;
        }
    }
    return 0;
}

关于这题

思路：把点带到给定参数的直线方程中 与0比较 确定是A类还是B类
这里用一个vector 会简便很多
可以将vector理解为动态数组，是封装好了的类
进行vector操作前应添加头文件#include <vector>
这里简单介绍几种常用的用法

 #include<vector>
vector<int> a,b;
//a含有4个值为2的元素
a.assign(4,2);
//返回a的最后一个元素
a.back();
//返回a的第一个元素
a.front();
//清空a中的元素
a.clear();
//判断a是否为空，空则返回true，非空则返回false
a.empty();
//删除a向量的最后一个元素
a.pop_back();
//在a的最后一个向量后插入一个元素，其值为5
a.push_back(5);
//返回a中元素的个数
a.size();
//返回a在内存中总共可以容纳的元素个数
a.capacity();
//将a的现有元素个数调整至10个，多则删，少则补，其值随机
a.resize(10);
//将a的现有元素个数调整至10个，多则删，少则补，其值为2
a.resize(10,2);
//将a的容量扩充至100，
a.reserve(100);
//b为向量，将a中的元素和b中的元素整体交换
a.swap(b);
//从向量中读取元素
//通过下标方式获取
int a[6]={1,2,3,4,5,6};
vector<int>b(a,a+4);
for(int i=0;i<=b.size()-1;++i){cout<<b[i]<<endl;}