CCF CSP 202006-1 线性分类器

方法1

数组方法

#include <iostream>
using namespace std;

int main(int argc, const char * argv[]) {
    int n=0,m=0;
    cin>>n>>m;
    int* left=new int [n];
    int* right=new int [n];
    char* type=new char [n];
    for(int i=0;i<n;i++){
        cin>>left[i]>>right[i]>>type[i];
    }
    for(int j=0;j<m;j++){
        int a=0,b=0,c=0;
        char t=0;
        string output="Yes";
        cin>>a>>b>>c;
        for(int k=0;k<n;k++){
            int res=a+b*left[k]+c*right[k];
            if(res==0){
                output="No";
                break;
            }
            if(k==0){
                if(res<0){
                    if(type[0]=='A')
                        t='B';
                    else
                        t='A';
                }
                if(res>0)
                    t=type[0];
            }
            else{
                if(type[k]==t&&res<0){
                    output="No";
                    break;
                }
                if(type[k]!=t&&res>0){
                    output="No";
                    break;
                }
            }
        }
        cout<<output<<endl;
    }
    return 0;
}

方法2

Vector方法

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

int main(int argc, const char * argv[]) {
     int n=0,m=0;
     cin>>n>>m;
     int* left=new int [n];
     int* right=new int [n];
     char* type=new char [n];
     for(int i=0;i<n;i++){
         cin>>left[i]>>right[i]>>type[i];
     }
     for(int j=0;j<m;j++){
         int a=0,b=0,c=0;
         vector<char> up;
         vector<char> down;
         string output="Yes";
         int flag=1;//判断点是否在线上
         cin>>a>>b>>c;
         for(int k=0;k<n;k++){
             
             int res=a+b*left[k]+c*right[k];
             if(res==0){
                 output="No";
                 flag=0;//点在线上
                 break;
             }
             else if(res>0)
                 up.push_back(type[k]);
             else
                 down.push_back(type[k]);
         }
         if(!flag) continue;
         for(int i=1;i<up.size();i++)
             if(up[i]!=up[i-1]){
                 output="No";
                 break;
             }
         for(int i=1;i<down.size();i++)
             if(down[i]!=down[i-1]){
             output="No";
             break;
             }
         cout<<output<<endl;
     }
     return 0;
 }