线性分类器

CCF真题

 

 

#include<iostream>
#include<vector>
#include<climits>
#include<cstdbool>
#include<algorithm>
#include<set>
#include<map>
using namespace std;
string YesOrNo(vector<int>&l,vector<vector<int>>&s,vector<char>&c)
{
    string res="Yes";
    int lastA=0,lastB=0;
    for(int i=0; i<s.size(); i++)
    {
        if(c[i]=='A')
        {
            int sum=l[0]+l[1]*s[i][0]+l[2]*s[i][1];
            if(i==0)
            {
                if(sum>0)
                {
                    lastA=1;
                    lastB=-1;
                }
                if(sum<0)
                {
                    lastA=-1;
                    lastB=1;
                }
            }
            else
            {
                if((sum>0&&lastA==-1)||(sum<0&&lastA==1))
                {
                    res="No";
                    break;
                }
            }
        }
        else
        {
            int sum=l[0]+l[1]*s[i][0]+l[2]*s[i][1];
            if(i==0)
            {
                if(sum<0)
                {
                    lastA=1;
                    lastB=-1;
                }
                if(sum>0)
                {
                    lastA=-1;
                    lastB=1;
                }
            }
            else
            {
                if((sum>0&&lastB==-1)||(sum<0&&lastB==1))
                {
                    res="No";
                    break;
                }
            }
        }
    }
    return res;
}
int main()
{
    int m,n;
    cin>>m>>n;
    vector<char>c(m);
    vector<vector<int>>s(m,vector<int>(2));
    vector<vector<int>>l(n,vector<int>(3));
    for(int i=0; i<m; i++)
    {
        cin>>s[i][0]>>s[i][1];
        cin>>c[i];
    }
    for(int i=0; i<n; i++)
    {
        cin>>l[i][0]>>l[i][1]>>l[i][2];
    }
    for(int i=0;i<n;i++)
    {
        cout<<YesOrNo(l[i],s,c)<<endl;
    }
    return 0;
}

 注意输出格式的问题。