牛客周赛 Round 57 D-小红的线段（贪心）

 示例1

输入

2 -1 2
0 0
0 1
1 0
1 1

输出

1
1 2 N
3 4 Y

说明

连接第一个点和第二个点，和直线没有交点。连接第三个点和第四个点，和直线有交点。

 

贪心策略：

把点集分为三部分，直线上方m1、直线下方m2以及在直线上m3，我们可以发现：

m1中的点和m2中的任意点相连都能通过直线；m3中的点由于在直线上，与m1，m2，m3中的任意点相连都能通过直线。显然m1和m2的匹配条件更为苛刻。

1.把m1和m2中的点两两进行匹配

2.把策略1中匹配剩下的点（若有）与m3中的点两两进行匹配

3.把m3中剩下的点（若有）两两进行匹配

至此所有能通过直线的点都匹配完成

#include<bits/stdc++.h>
using namespace std;
int n,k,b;
vector<int> m1,m2,m3; //记录点集，m1中是直线上方的点，m2中是直线下方的点，m3中是在直线上的点
map<pair<int,int>, char> result; //记录结果
int cnt = 0; //记录相交线段数
void solve(vector<int> m1, vector<int> m2, vector<int> m3){ //m1是上下点集中数量少的，m2是数量大的，m3是直线上的点集
    auto it1 = m1.begin(), it2 = m2.begin(), it3 = m3.begin();
    //匹配m1与m2
    for (;it1 != m1.end(); ++it1, ++it2){ 
        result.insert(make_pair(make_pair(*it1, *it2), 'Y'));
        cnt++;
    }
    //m2中剩余的与m3进行匹配
    if (m2.size() - m1.size() >= m3.size()){ 
        for (; it3 != m3.end(); ++it3, ++it2){
            result.insert(make_pair(make_pair(*it2, *it3), 'Y'));
            cnt++;
        }
        for (;it2 != m2.end(); ++it2){
            int temp = *it2;
            ++it2;
            result.insert(make_pair(make_pair(temp, *it2), 'N'));
        }
    }else{
        for (; it2 != m2.end(); ++it2, ++it3){
            result.insert(make_pair(make_pair(*it2, *it3), 'Y'));
            cnt++;
        }
        // m3中剩余的进行匹配
        for (;it3 != m3.end(); ++it3){ 
            int temp = *it3;
            ++it3;
            result.insert(make_pair(make_pair(temp, *it3), 'Y'));
            cnt++;
        }
    }
}
int main(){
    int x,y;
    cin>>n>>k>>b;
    for (int i = 1; i <= 2 * n; ++i){
        cin>>x>>y;
        if (y < k * x + b){
            m1.push_back(i);
        }else if (y == k * x + b){
            m3.push_back(i);
        }else{
            m2.push_back(i);
        }
    }
    if (m1.size() > m2.size()){
        solve(m2,m1,m3);
    }else{
        solve(m1,m2,m3);
    }
    cout<<cnt<<endl;
    for (auto i : result){
        cout<<i.first.first<<" "<<i.first.second<<" "<<i.second<<endl;
    }
}