[프로그래머스] 체육복

링크

https://programmers.co.kr/learn/courses/30/lessons/42862

코딩테스트 연습 - 체육복 | 프로그래머스

풀이

 

 여벌 체육복이 있는 학생이 이들에게 체육복을 빌려주려 하고. 학생들의 번호는 체격 순으로 매겨져 있어, 바로 앞번호의 학생이나 바로 뒷번호의 학생에게만 체육복을 빌려줄 수 있는 문제 인데

 

저는 여벌의 체육복을 가지고 있는 학생의 양 옆에 체육복이 없는 학생들이 있을때 무조건 앞에 있는 친구부터 주도록 하였습니다. 

 

앞에 학생은 체육복이 있고 뒤에 있는 학생은 체육복이 없을때는 이때는 뒤에 학생을 주도록 하였습니다.

 

학생이 가지고 있는 체육복의 갯수를 확인하기 위하여 배열을 만들어 1로 초기화 하였고

 

잃은 학생은 -1 , 여분을 가지고 있는 학생은 +1 해주어 체육복 갯수 배열을 완성하였고 

 

학생을 기준으로 앞 뒤를 확인 하여 계산 하였습니다.

 

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<cstring>
#include<functional>
#include<cmath>

#define SIZE 40
const int INF = 2000000000;

using namespace std;

typedef long long int ll;

int check[SIZE];
int solution(int n, vector<int> lost, vector<int> reserve) {
	for (int i = 0; i < SIZE; i++) {
		check[i] = 1;
	}

	for (auto l : lost) {
		check[l]--;
	}
	for (auto r : reserve) {
		check[r]++;
	}

	int answer = 0;

	for (int i = 1; i <= n; i++) {
		if (check[i] == 2) {
			if (check[i - 1] == 0) {
				check[i]--; check[i - 1]++;
			}
			else if (check[i + 1] == 0) {
				check[i]--; check[i + 1]++;
			}
		}
	}

	for (int i = 1; i <= n; i++) {
		if (check[i])
			answer++;
	}
	return answer;
}