백준 1337번 올바른 배열 C++

출처 : https://www.acmicpc.net/problem/1337

 

배열을 정렬하고 탐색하면서 배열의 원소 + 4 보다 작거나 같은 원소의 개수를 찾으면서 문제를 해결하였다.

 

#include <iostream>
#include <string>
#include <vector>
#include <math.h>
#include <algorithm>
#include <utility>
#include <stack>
#include <queue>
#include <math.h>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#include <iomanip>

using namespace std;
using ll = long long;

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int n;
	cin >> n;
	vector<int> v(n);
	for (int i = 0; i < n; i++) {
		cin >> v[i];
	}
	sort(v.begin(), v.end());
	int ans = 5;
	for (int i = 0; i < n; i++) {
		int target = v[i] + 4;
		int idx = i + 1;
		int cnt = 1;
		while (idx < n && v[idx] <= target) {
			cnt++;
			idx++;
		}
		ans = min(ans, 5 - cnt);
	}
	cout << ans;
};