배열을 다루는 알고리즘 문제를 해결해본다.
모든 코드는 깃허브 (링크)의 테스트 코드로 정리해두었다.
큰 수 출력하기
N(1<=N<=100)개의 정수를 입력받아, 자신의 바로 앞 수보다 큰 수만 출력하는 프로그램을 작 성하세요.(첫 번째 수는 무조건 출력한다)
- 입력설명
첫 줄에 자연수 N이 주어지고, 그 다음 줄에 N개의 정수가 입력된다. - 출력설명
자신의 바로 앞 수보다 큰 수만 한 줄로 출력한다. - 입력예제 1
6
7 3 9 5 6 12 - 출력예제 1
7 9 6 12
public class PrintBigNumber {
public Integer[] solution1(int... inputs) {
List<Integer> answer = new ArrayList<>(List.of(inputs[0]));
for (int i = 1; i < inputs.length; i++) {
if (inputs[i - 1] < inputs[i]) {
answer.add(inputs[i]);
}
}
return answer.toArray(new Integer[0]);
}
@Test
@DisplayName("큰 수 출력하기")
public void main() {
Integer[] expectedAnswer = {7, 9, 6, 12};
Integer[] answer1 = solution1(7, 3, 9, 5, 6, 12);
assertArrayEquals(expectedAnswer, answer1);
}
}보이는 학생
선생님이 N명의 학생을 일렬로 세웠습니다. 일렬로 서 있는 학생의 키가 앞에서부터 순서대로 주어질 때,
맨 앞에 서 있는 선생님이 볼 수 있는 학생의 수를 구하는 프로그램을 작성하세요.
(어떤 학생이 자기 앞에 서 있는 학생들보다 크면 그 학생은 보이고, 작거나 같으면 보이지 않 습니다.)
- 입력설명
첫 줄에 정수 N(5<=N<=100,000)이 입력된다. 그 다음줄에 N명의 학생의 키가 앞에서부터 순 서대로 주어진다. - 출력설명
선생님이 볼 수 있는 최대학생수를 출력한다. - 입력예제 1
8
130 135 148 140 145 150 150 153 - 출력예제 1
5
public class VisibleStudent {
public int solution1(int... inputs) {
int visibleCount = 1;
int maxHeight = inputs[0];
for (int i = 1; i < inputs.length; i++) {
if (maxHeight < inputs[i]) {
visibleCount++;
maxHeight = inputs[i];
}
}
return visibleCount;
}
@Test
@DisplayName("보이는 학생")
public void main() {
int expectedAnswer = 5;
int answer1 = solution1(130, 135, 148, 140, 145, 150, 150, 153);
assertEquals(answer1, expectedAnswer);
}
}가위 바위 보
A, B 두 사람이 가위바위보 게임을 합니다. 총 N번의 게임을 하여 A가 이기면 A를 출력하고, B가 이기면 B를 출력합니다. 비길 경우에는 D를 출력합니다.
가위, 바위, 보의 정보는 1:가위, 2:바위, 3:보로 정하겠습니다.
예를 들어 N=5이면
회수 1 2 3 4 5
A의 정보 2 3 3 1 3
B의 정보 1 1 2 2 3
승자 A B A B D
두 사람의 각 회의 가위, 바위, 보 정보가 주어지면 각 회를 누가 이겼는지 출력하는 프로그램 을 작성하세요.
- 입력설명
첫 번째 줄에 게임 횟수인 자연수 N(1<=N<=100)이 주어집니다. 두 번째 줄에는 A가 낸 가위, 바위, 보 정보가 N개 주어집니다. 세 번째 줄에는 B가 낸 가위, 바위, 보 정보가 N개 주어집니다. - 출력설명
각 줄에 각 회의 승자를 출력합니다. 비겼을 경우는 D를 출력합니다.
입력예제 1
5
2 3 3 1 3
1 1 2 2 3 - 출력예제 1
A B A B D
public class RockScissorsPaper {
public String[] solution1(int[] a, int[] b) {
List<String> answer = new ArrayList<>();
for (int i = 0; i < a.length; i++) {
if (a[i] == b[i]) {
answer.add("D");
} else if (a[i] == 1 && b[i] == 3) {
answer.add("A");
} else if (a[i] == 2 && b[i] == 1) {
answer.add("A");
} else if (a[i] == 3 && b[i] == 2) {
answer.add("A");
} else {
answer.add("B");
}
}
return answer.toArray(new String[0]);
}
@Test
@DisplayName("가위 바위 보")
public void main() {
String[] expectedAnswer = {"A", "B", "A", "B", "D"};
String[] answer1 = solution1(new int[]{2, 3, 3, 1, 3}, new int[]{1, 1, 2, 2, 3});
Assertions.assertArrayEquals(answer1, expectedAnswer);
}
}피보나치 수열
1) 피보나키 수열을 출력한다. 피보나치 수열이란 앞의 2개의 수를 합하여 다음 숫자가 되는 수열이다.
2) 입력은 피보나치 수열의 총 항의 수 이다. 만약 7이 입력되면 1 1 2 3 5 8 13을 출력하면 된다.
- 입력설명
첫 줄에 총 항수 N(3<=N<=45)이 입력된다. - 출력설명
첫 줄에 피보나치 수열을 출력합니다. - 입력예제 1
10 - 출력예제 1
1 1 2 3 5 8 13 21 34 55
public class FibonacciNumbers {
public int[] solution1(int input) {
int[] answer = new int[input];
for (int i = 0; i < input; i++) {
if (i <= 1) {
answer[i] = 1;
} else {
answer[i] = (answer[i - 2] + answer[i - 1]);
}
}
return answer;
}
@Test
@DisplayName("피보나치 수열")
public void main() {
int[] expectedAnswer = {1, 1, 2, 3, 5, 8, 13, 21, 34, 55};
int[] answer = solution1(10);
assertArrayEquals(answer, expectedAnswer);
}
}소수 (에라토스테네스 체)
자연수 N이 입력되면 1부터 N까지의 소수의 개수를 출력하는 프로그램을 작성하세요.
만약 20이 입력되면 1부터 20까지의 소수는 2, 3, 5, 7, 11, 13, 17, 19로 총 8개입니다. 제한시간은 1초입니다.
- 입력설명
첫 줄에 자연수의 개수 N(2<=N<=200,000)이 주어집니다. - 출력설명
첫 줄에 소수의 개수를 출력합니다. - 입력예제 1
20 - 출력예제 1
8
public class PrimeNumber {
public int solution1(int input) {
int count = 0;
int[] numbers = new int[input + 1];
for (int i = 2; i < input; i++) {
if (numbers[i] == 0) {
count++;
for (int j = i; j < input; j = j + i) {
numbers[j] = 1;
}
}
}
return count;
}
@Test
@DisplayName("소수 (에라토스테네스 체)")
public void main() {
int expectedCount = 8;
int answer1 = solution1(20);
assertEquals(expectedCount, answer1);
}
}뒤집은 소수
N개의 자연수가 입력되면 각 자연수를 뒤집은 후 그 뒤집은 수가 소수이면 그 소수를 출력하는 프로그램을 작성하세요.
예를 들어 32를 뒤집으면 23이고, 23은 소수이다. 그러면 23을 출 력한다. 단 910를 뒤집으면 19로 숫자화 해야 한다.
첫 자리부터의 연속된 0은 무시한다.
- 입력설명
첫 줄에 자연수의 개수 N(3<=N<=100)이 주어지고, 그 다음 줄에 N개의 자연수가 주어진다. 각 자연수의 크기는 100,000를 넘지 않는다. - 출력설명
첫 줄에 뒤집은 소수를 출력합니다. 출력순서는 입력된 순서대로 출력합니다. - 입력예제 1
9
32 55 62 20 250 370 200 30 100 - 출력예제 1
23 2 73 2 3
public class FlipPrimeNumber {
public boolean isPrime(int number) {
if (number <= 1) {
return false;
}
for (int i = 2; i < number; i++) {
if (number % i == 0) {
return false;
}
}
return true;
}
public Integer[] solution1(int... inputs) {
List<Integer> flipPrimeNumbers = new ArrayList<>();
for (int i = 0; i < inputs.length; i++) {
int tmp = inputs[i];
int result = 0;
while (tmp > 0) {
// 입력받은 값을 10으로 나눈 값의 소수점을 구한다.
int decimalPoint = tmp % 10;
// 결과 값에 10을 곱해주고 소수점을 더한다.
result = result * 10 + decimalPoint;
// 현재 값을 10으로 나누어 1의 자리를 버리도록 한다.
tmp = tmp / 10;
}
if (isPrime(result)) {
flipPrimeNumbers.add(result);
}
}
return flipPrimeNumbers.toArray(new Integer[0]);
}
// StringBuilder의 reverse를 활용한 방법이다.
// 입력받은 문자열을 String으로 변환하여 reverse함수를 사용하여 뒤집는다.
// 뒤집은 문자열을 Integer의 parseInt 함수를 사용하여 형변환을 시킨다.
// 이 때 가장 앞자리가 0인 경우 0은 제거되고 숫자로 형변환이 된다.
public Integer[] solution2(int... inputs) {
List<Integer> flipPrimeNumbers = new ArrayList<>();
for (int input : inputs) {
input = Integer.parseInt(new StringBuilder(String.valueOf(input)).reverse().toString());
if (isPrime(input)) {
flipPrimeNumbers.add(input);
}
}
return flipPrimeNumbers.toArray(new Integer[0]);
}
@Test
@DisplayName("뒤집은 소수")
public void main() {
Integer[] expectedAnswer = {23, 2, 73, 2, 3};
Integer[] answer1 = solution1(32, 55, 62, 20, 250, 370, 200, 30, 100);
assertArrayEquals(expectedAnswer, answer1);
Integer[] answer2 = solution2(32, 55, 62, 20, 250, 370, 200, 30, 100);
assertArrayEquals(expectedAnswer, answer2);
}
}점수 계산
OX 문제는 맞거나 틀린 두 경우의 답을 가지는 문제를 말한다.
여러 개의 OX 문제로 만들어진 시험에서 연속적으로 답을 맞히는 경우에는 가산점을 주기 위해서 다음과 같이 점수 계산을 하기로 하였다.
1번 문제가 맞는 경우에는 1점으로 계산한다. 앞의 문제에 대해서는 답을 틀리다가 답이 맞는 처음 문제는 1점으로 계산한다.
또한, 연속으로 문제의 답이 맞는 경우에서 두 번째 문제는 2점, 세 번째 문제는 3점, ..., K번째 문제는 K점으로 계산한다.
틀린 문제는 0점으로 계산한다.
예를 들어, 아래와 같이 10 개의 OX 문제에서 답이 맞은 문제의 경우에는 1로 표시하고, 틀린 경우에는 0으로 표시하였을 때,
점수 계산은 아래 표와 같이 계산되어, 총 점수는 1+1+2+3+1+2=10 점이다.
1011100110
시험문제의 채점 결과가 주어졌을 때, 총 점수를 계산하는 프로그램을 작성하시오.
- 입력설명
첫째 줄에 문제의 개수 N (1 ≤ N ≤ 100)이 주어진다. 둘째 줄에는 N개 문제의 채점 결과를 나 타내는 0 혹은 1이 빈 칸을 사이에 두고 주어진다. 0은 문제의 답이 틀린 경우이고, 1은 문제의 답이 맞는 경우이다. - 출력설명
첫째 줄에 입력에서 주어진 채점 결과에 대하여 가산점을 고려한 총 점수를 출력한다. - 입력예제 1
10 1011100110 - 출력예제 1
10
public class CalculateScore {
public int solution1(int[] inputs) {
int totalScore = 0;
int accumulatedScore = 1;
for (int i = 0; i < inputs.length; i++) {
if (inputs[i] == 1) {
totalScore += accumulatedScore;
accumulatedScore++;
} else {
accumulatedScore = 1;
}
}
return totalScore;
}
@Test
@DisplayName("점수 계산")
public void main() {
int expectedAnswer = 10;
int answer1 = solution1(new int[]{1, 0, 1, 1, 1, 0, 0, 1, 1, 0});
assertEquals(expectedAnswer, answer1);
}
}등수구하기
N명의 학생의 국어점수가 입력되면 각 학생의 등수를 입력된 순서대로 출력하는 프로그램을 작성하세요.
같은 점수가 입력될 경우 높은 등수로 동일 처리한다.
즉 가장 높은 점수가 92점인데 92점이 3명 존재하면 1등이 3명이고 그 다음 학생은 4등이 된다.
- 입력설명
첫 줄에 N(3<=N<=100)이 입력되고, 두 번째 줄에 국어점수를 의미하는 N개의 정수가 입력 된다. - 출력설명
입력된 순서대로 등수를 출력한다. - 입력예제 1
5
87 89 92 100 76 - 출력예제 1
4 3 2 1 5
public class GetRanking {
public Integer[] solution1(int[] scores) {
List<Integer> rankings = new ArrayList<>();
int rank = 1;
for (int score: scores) {
for (int i = 0; i < scores.length; i++) {
if (score < scores[i]) {
rank++;
}
}
rankings.add(rank);
rank = 1;
}
return rankings.toArray(new Integer[0]);
}
@Test
@DisplayName("등수 구하기")
public void main() {
Integer[] expectedAnswer = {4, 3, 2, 1, 5};
Integer[] answer1 = solution1(new int[]{87, 89, 92, 100, 76});
assertArrayEquals(expectedAnswer, answer1);
}
}격자판 최대합
55 격자판에 아래롸 같이 숫자가 적혀있습니다.
10 13 10 12 15
12 39 30 23 11
11 25 50 53 15
19 27 29 37 27
19 13 30 13 19
NN의 격자판이 주어지면 각 행의 합, 각 열의 합, 두 대각선의 합 중 가 장 큰 합을 출력합 니다.
- 입력설명
첫 줄에 자연수 N이 주어진다.(2<=N<=50)
두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다. 각 자연수는 100을 넘지 않는 다. - 출력설명 최대합을 출력합니다.
- 입력예제 1
5
10 13 10 12 15
12 39 30 23 11
11 25 50 53 15
19 27 29 37 27
19 13 30 13 19 - 출력예제 1
155
public class GridMaxSummary {
public int solution1(int[][] rows) {
int maxSummary = Integer.MIN_VALUE;
int rowCount = rows.length;
for (int[] row : rows) {
int tmpSum1 = Arrays.stream(row).sum();
int tmpSum2 = 0;
for (int j = 0; j < rowCount; j++) {
tmpSum2 += row[j];
}
if (Math.max(tmpSum1, tmpSum2) > maxSummary) {
maxSummary = Math.max(tmpSum1, tmpSum2);
}
}
int tmpSum3 = 0;
int tmpSum4 = 0;
for (int i = 0; i < rowCount; i++) {
tmpSum3 += rows[i][i];
tmpSum4 += rows[i][rowCount - i - 1];
}
if (Math.max(tmpSum3, tmpSum4) > maxSummary) {
maxSummary = Math.max(tmpSum3, tmpSum4);
}
return maxSummary;
}
@Test
@DisplayName("격자판 최대합")
public void main() {
int expectedAnswer = 155;
int[][] inputs1 = {
{10, 13, 10, 12, 15},
{12, 39, 30, 23, 11},
{11, 25, 50, 53, 15},
{19, 27, 29, 37, 27},
{19, 13, 30, 13, 19}
};
int answer1 = solution1(inputs1);
Assertions.assertEquals(expectedAnswer, answer1);
}
}봉우리
지도 정보가 N*N 격자판에 주어집니다. 각 격자에는 그 지역의 높이가 쓰여있습니다.
각 격자 판의 숫자 중 자신의 상하좌우 숫자보다 큰 숫자는 봉우리 지역입니다.
봉우리 지역이 몇 개 있는지 알아내는 프로그램을 작성하세요.
격자의 가장자리는 0으로 초기화 되었다고 가정한다.
만약 N=5 이고, 격자판의 숫자가 다음과 같다면 봉우리의 개수는 10개입니다.
0 0 0 0 0 0 0
0 (5) 3 (7) 2 (3) 0
0 3 (7) 1 (6) 1 0
0 (7) 2 5 3 (4) 0
0 4 3 (6) 4 1 0
0 (8) 7 3 (5) 2 0
0 0 0 0 0 0 0
- 입력설명
첫 줄에 자연수 N이 주어진다.(2<=N<=50)
두 번째 줄부터 N줄에 걸쳐 각 줄에 N개의 자연수가 주어진다. 각 자연수는 100을 넘지 않는다. - 출력설명
봉우리의 개수를 출력하세요. - 입력예제 1
5
53723
37161
72534
43641
87352 - 출력예제 1
10
public class MountainPeak {
public int solution1(int[][] inputs) {
int[] xDirection = {-1, 0, 1, 0};
int[] yDirection = {0, 1, 0, -1};
int peakCount = 0;
for (int i = 0; i < inputs.length; i++) {
for (int j = 0; j < inputs.length; j++) {
boolean flag = true;
for (int k = 0; k < 4; k++) {
int nextX = i + xDirection[k];
int nextY = j + yDirection[k];
// nextX와 nextY가 음수 또는 배열 범위 밖의 값을 참조하는 것을 방지한다.
if (nextX >= 0 && nextX < inputs.length
&& nextY >= 0 && nextY < inputs.length
&& inputs[nextX][nextY] >= inputs[i][j]) {
flag = false;
break;
}
}
if (Boolean.TRUE == flag) {
peakCount++;
}
}
}
return peakCount;
}
@Test
@DisplayName("봉우리")
public void main() {
int[][] inputs = {
{5, 3, 7, 2, 3},
{3, 7, 1, 6, 1},
{7, 2, 5, 3, 4},
{4, 3, 6, 4, 1},
{8, 7, 3, 5, 2}
};
int expectedAnswer = 10;
int answer1 = solution1(inputs);
assertEquals(expectedAnswer, answer1);
}
}임시 반장 정하기
김갑동 선생님은 올해 6학년 1반 담임을 맡게 되었다.
김갑동 선생님은 우선 임시로 반장을 정하고 학생들이 서로 친숙해진 후에 정식으로 선거를 통해 반장을 선출하려고 한다.
그는 자기반 학생 중에서 1학년부터 5학년까지 지내오면서 한번이라도 같은 반이었던 사람이 가장 많은 학생을 임시 반장으로 정하려 한다.
그래서 김갑동 선생님은 각 학생들이 1학년부터 5학년까지 몇 반에 속했었는지를 나타내는 표를 만들었다.
예를 들어 학생 수가 5명일 때의 표를 살펴보자.
1학년 2학년 3학년 4학년 5학년
1번 학생 2 3 1 7 3
2번 학생 4 1 9 6 8
3번 학생 5 5 2 4 4
4번 학생 6 5 2 6 7
5번 학생 8 4 2 2 2
위 경우에 4번 학생을 보면 3번 학생과 2학년 때 같은 반이었고, 3번 학생 및 5번 학생과 3학 년 때 같은 반이었으며,
2번 학생과는 4학년 때 같은 반이었음을 알 수 있다.
그러므로 이 학급에서 4번 학생과 한번이라도 같은 반이었던 사람은 2번 학생, 3번 학생과 5번 학생으로 모두 3명이다.
이 예에서 4번 학생이 전체 학생 중에서 같은 반이었던 학생 수가 제일 많으므로 된다.
1학년부터 5학년까지 속했던 반이 주어질 때, 임시 반장을 정하는 프로그램을 작성하시오.
- 입력설명
첫째 줄에는 반의 학생 수를 나타내는 정수가 주어진다. 학생 수는 3 이상 1000이하이다. 둘째 줄부터는 1번 학생부터
차례대로 각 줄마다 1학년부터 5학년까지 몇 반에 속했었는지를 나타내는 5개의 정수가 빈칸 하나를 두고 주어진다.
주어지는 정수는 모두 1 이상 9 이하의 정수이다. - 출력설명
첫 줄에 임시 반장으로 정해진 학생의 번호를 출력한다.
단, 임시 반장이 될 수 있는 학생이 여러 명인 경우에는 그 중 가장 작은 번호만 출력한다. - 입력예제 1
23173
41968
55244
65267
84222 - 출력예제 1
4
public class TemporaryClassPresident {
public int solution1(int[][] inputs) {
int studentCount = inputs.length;
int maxFriendship = Integer.MIN_VALUE;
int presidentNumber = 0;
for (int i = 0; i < studentCount; i++) {
int currentFriendship = Integer.MIN_VALUE;
for (int j = 0; j < studentCount; j++) {
// 5학년까지 존재하므로
for (int k = 0; k < 5; k++) {
if (inputs[i][k] == inputs[j][k]) {
currentFriendship++;
}
}
}
if (currentFriendship > maxFriendship) {
presidentNumber = i;
maxFriendship = currentFriendship;
}
}
// n번 학생은 n - 1번 인덱스에 정보가 있으므로
return presidentNumber + 1;
}
@Test
@DisplayName("임시반장 정하기")
public void main() {
int[][] inputs = {
{2, 3, 1, 7, 3},
{4, 1, 9, 6, 8},
{5, 5, 2, 4, 4},
{6, 5, 2, 6, 7},
{8, 4, 2, 2, 2}
};
int expectedAnswer = 4;
int answer1 = solution1(inputs);
assertEquals(expectedAnswer, answer1);
}
}멘토링
현수네 반 선생님은 반 학생들의 수학점수를 향상시키기 위해 멘토링 시스템을 만들려고 합니다.
멘토링은 멘토(도와주는 학생)와 멘티(도움을 받는 학생)가 한 짝이 되어 멘토가 멘티의 수학공부를 도와주는 것입니다.
선생님은 M번의 수학테스트 등수를 가지고 멘토와 멘티를 정합니다.
만약 A학생이 멘토이고, B학생이 멘티가 되는 짝이 되었다면, A학생은 M번의 수학테스트에서 모두 B학생보다 등수가 앞서야 합니다.
M번의 수학성적이 주어지면 멘토와 멘티가 되는 짝을 만들 수 있는 경우가 총 몇 가지 인지 출력하는 프로그램을 작성하세요.
- 입력설명
첫 번째 줄에 반 학생 수 N(1<=N<=20)과 M(1<=M<=10)이 주어진다.
두 번째 줄부터 M개의 줄에 걸쳐 수학테스트 결과가 학생번호로 주어진다.
학생번호가 제일 앞에서부터 1등, 2등, ...N등 순으로 표현된다.
만약 한 줄에 N=4이고, 테스트 결과가 3 4 1 2로 입력되었다면 3번 학생이 1등, 4번 학생이 2등,
1번 학생이 3등, 2번 학생이 4등을 의미합니다. - 출력설명
첫 번째 줄에 짝을 만들 수 있는 총 경우를 출력합니다. - 입력예제 1
4 3
3 4 1 2
4 3 2 1
3 1 4 2 - 출력예제 1
3
(3, 1), (3, 2), (4, 2)와 같이 3가지 경우의 (멘토, 멘티) 짝을 만들 수 있다.
public class Mentoring {
public int solution1(int studentCount, int testCount, int[][] scores) {
int answer = 0;
for (int i = 1; i < studentCount; i++) {
for (int j = 1; j < studentCount; j++) {
int count = 0;
for (int k = 0; k < testCount; k++) {
int iPosition = 0;
int jPosition = 0;
for (int s = 0; s < studentCount; s++) {
if (scores[k][s] == i) {
iPosition = s;
}
if (scores[k][s] == j) {
jPosition = s;
}
if (iPosition < jPosition) {
count++;
}
}
if (count == testCount) {
answer++;
}
}
}
}
return answer;
}
@Test
@DisplayName("멘토링")
public void main() {
int[][] scores = {
{3, 4, 1, 2},
{4, 3, 2, 1},
{3, 1, 4, 2}
};
int expectedAnswer = 3;
int answer1 = solution1(4, 3, scores);
Assertions.assertEquals(expectedAnswer, answer1);
}
}'Algorithm' 카테고리의 다른 글
| [Algorithm] Stack (0) | 2022.02.23 |
|---|---|
| [Algorithm] HashMap (0) | 2022.02.22 |
| [Algorithm] Sliding window (0) | 2022.02.21 |
| [Algorithm] Two Pointers (0) | 2022.02.21 |
| [Algorithm] 문자열 다루기 (0) | 2022.02.20 |
