[C++] 백준(BOJ)-15975번 화살표 그리기

https://www.acmicpc.net/problem/15975

15975번: 화살표 그리기

#include <algorithm>
#include <iostream>
#include <vector>
#include <set>
using namespace std;
bool Compare(pair<int,int> a,pair<int,int> b) {
  if(a.second==b.second){
    return a.first < b.first;
  }
  else{
    return a.second < b.second;
  }
};
int main()
{
  long long n=0;
	vector<pair<long long,long long>> v;
  vector<long long> v1;
	cin >> n;
  long long temp1=0;
  long long temp2=0;
  long long count=0;
	for (int i = 0; i < n; i++) {
    cin >> temp1 >> temp2;
    v.push_back(make_pair(temp1,temp2));
	}
  sort(v.begin(),v.end(), Compare);
  
  if(v[0].second == v[1].second){
    count = v[1].first - v[0].first;
  }

  for(int i=1;i<n-1;i++){
    if(v[i].second != v[i+1].second && v[i-1].second != v[i].second){
      continue;
    }
    if(v[i].second != v[i+1].second){
      count += v[i].first - v[i-1].first;
      continue;
    }
    if(v[i-1].second != v[i].second){
      count += v[i+1].first - v[i].first;
      continue;
    }
    temp1 = v[i+1].first - v[i].first;
    temp2 = v[i].first - v[i-1].first;
    if(temp1 <= temp2){
      count += temp1;
    }
    else{
      count += temp2;
    }
  }
  if(v[n-1].second == v[n-2].second){
    count += v[n-1].first - v[n-2].first;
  }
  cout << count;
}

long long으로 type을 바꿨더니 바로 풀렸다.

그 외에는 compare 함수 써서 sort 하기, 그 다음에 비교하여 count 값에 더하기 정도의 방법으로 풀었다.