寝癖頭の解法

小学生の目線から、勉強中の覚え書きを投稿、更新していきます。

Aizu Online Judge in C++ #CGL_6_A : Segment Intersections: Manhattan Geometry

Aizu Online Judge(AOJ)の過去問から、その提出コードの解答例です。

・問題 "Segment Intersections: Manhattan Geometry"
https://onlinejudge.u-aizu.ac.jp/problems/CGL_6_A
・線分交差(マンハッタン幾何)

僕が作成、提出したコードは、以下のとおりです。

Aizu Online Judge in C++ #CGL_6_A : Segment Intersections: Manhattan Geometry
/*
Aizu Online Judge in C++ #CGL_6_A : Segment Intersections: Manhattan Geometry
https://onlinejudge.u-aizu.ac.jp/problems/CGL_6_A
 提出コードの解答例
 https://neguse-atama.hatenablog.com
*/
#include<bits/stdc++.h>
using namespace std;
int main(void){
    vector<pair<int,pair<int,int>>> vx,vy;
    int i=0,r=0,n,x1,y1,x2,y2;
    for(cin>>n;i<n;i++){
        cin>>x1>>y1>>x2>>y2;
        if(x1>x2){
            swap(x1,x2);
        }
        if(y1>y2){
            swap(y1,y2);
        }
        if(x1==x2){
            vy.emplace_back(make_pair(x1,make_pair(y1,y2)));
        }else{
            vx.emplace_back(make_pair(y1,make_pair(x1,x2)));
        }
    }
    sort(vy.begin(),vy.end());
    for(auto &e : vx){
        auto it1=lower_bound(vy.begin(),vy.end(),make_pair(e.second.first,make_pair(-2000000000,-2000000000)));
        auto it2=lower_bound(vy.begin(),vy.end(),make_pair(e.second.second+1,make_pair(-2000000000,-2000000000)));
        for(;it1!=it2;++it1){
            auto &f=*it1;
            if(f.second.first<=e.first && e.first<=f.second.second){
                r++;
            }
        }
    }
    cout<<r<<endl;
    return 0;
}

設問の出典は、プログラミング問題のオンライン採点システム「Aizu Online Judge(AOJ)」です。
http://judge.u-aizu.ac.jp/onlinejudge/