기록

4번 최댓값

문제

9개의 서로 다른 자연수가 주어질 때, 이들 중 최댓값을 찾고 그 최댓값이 몇 번째 수인지를 구하는 프로그램을 작성하시오. 

예를 들어, 서로 다른 9개의 자연수 3, 29, 38, 12, 57, 74, 40, 85, 61 이 주어지면, 이들 중 최댓값은 85이고, 이 값은 8번째 수이다.

입력

첫째 줄부터 아홉 번째 줄까지 한 줄에 하나의 자연수가 주어진다. 주어지는 자연수는 100 보다 작다.

출력

첫째 줄에 최댓값을 출력하고, 둘째 줄에 최댓값이 몇 번째 수인지를 출력한다.

C++

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
    int n;
    vector<int> vec;
    while(cin >> n){
        vec.push_back(n);
    }
    auto iter = max_element(vec.begin(), vec.end());
    int index = distance(vec.begin(), iter);
    cout << *iter << "\n" << index + 1;
    return 0;
}

distance 함수를 사용해서 iter가 해당 벡터에서 몇 번째 인덱스인지 구한다.

C#

using System;
using System.Collections.Generic;
using System.Linq;
class Program{
    static void Main(string[] args){
        List<int> num_list = new List<int>();
        for (int i = 0; i < 9; i++){
            int val = int.Parse(Console.ReadLine());
            num_list.Add(val);
        }
        int max = num_list.Max();
        int index = num_list.IndexOf(max);
        Console.WriteLine($"{max}\n{index+1}");
    }
}

Python

num_list = []
for i in range(9):
    num_list.append(int(input()))
max_val = max(num_list)
index = num_list.index(max_val)
print(max_val)
print(index + 1)

[]로 가변 배열을 선언한다. 이 배열은 기본 타입인 list와 동일한 타입의 객체이다.

 num_list = list()와 동일한 객체

이 객체는 apeend를 사용해서 값을 추가할 수 있고 index로 요소의 인덱스를 검색할 수 있다.

Node.js

const fs = require('fs');
const num_list = fs.readFileSync('/dev/stdin', 'utf8').trim().split('\n').map(Number);
const max_num = Math.max(...num_list);
const index = num_list.indexOf(max_num);
console.log(max_num);
console.log(index + 1);

indexOf 메서드로 최댓값의 인덱스를 찾을 수 있다.

5번 공 넣기

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.

공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i j k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다. (1 ≤ i ≤ j ≤ N, 1 ≤ k ≤ N)

도현이는 입력으로 주어진 순서대로 공을 넣는다.

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.

나열된 바구니들을 배열 각 바구니들 인덱스로 보면 된다. 공에는 번호가 지정되어 있고 규칙에 맞춰 공을 바구니에 넣는다.

입력의 첫 줄에서 바구니의 개수 N과 바구니에 공을 담는 규칙 M개가 들어온다.

그다음에 M개의 줄에 각각 규칙이 들어오는데 순서대로 i j k 각 수는 i , j는 바구니의 범위, k는 넣을 공의 번호이다.

C++

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
    int n, m, i, j, k;
    cin >> n;
    cin >> m;
    vector<int> num_vec(n, 0);
    for (int l = 0; l < m; l++){
        cin >> i >> j >> k;
        fill(num_vec.begin() + i - 1, num_vec.begin() + j, k);
    }
     for (int num : num_vec) {
        cout << num << " ";
    }
    return 0;
}

컨테이너는 바구니의 개수만큼 사이즈를 만들고 값은 0으로 초기화해서 값이 없을 때 0으로 출력되도록 한다.

컨테이너의 범위 내에 특정 값을 채우기 위해서 fill 함수를 사용한다.

시작과 끝 범위, 채울 값을 인자로 받는다.

C#

using System;
using System.Collections.Generic;
using System.Linq;
class Program{
    static void Main(string[] args){
        int[] input = Console.ReadLine().Split().Select(int.Parse).ToArray();
        int n, m, i, j, k;
        n = input[0];
        m = input[1];
        List<int> num_list = new List<int>(new int[n]);
        for (int l = 0; l < m; l++){
            int[] rules = Console.ReadLine().Split().Select(int.Parse).ToArray();
            i = rules[0] - 1;
            j = rules[1] - 1;
            k = rules[2];
            for (int o = i; o <= j; o++){
                num_list[o] = k;
            }
        }
        
        foreach(int val in num_list){
            Console.Write($"{val} ");
        }
    }
}

리스트의 경우 특정 범위 내의 요소에 값을 넣을 때 사용할만한 메서드가 없어 이중 반복문으로 처리했다.

Python

input_val = list(map(int, input().split()))
n = input_val[0]
m = input_val[1]
num_list = [0] * n
for l in range(m):
    rules = list(map(int, input().split()))
    i = rules[0]
    j = rules[1]
    k = rules[2]
    for index in range(i - 1, j):
        num_list[index] = k
print(" ".join(map(str, num_list)))

Node.js

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin','utf8').split('\n');
const [n, m] = input[0].split(' ').map(Number);
let num_list = Array(n).fill(0);
for (let l = 1; l <= m; l++){
    const [i, j, k] = input[l].split(' ').map(Number);
    for (let idx = i - 1; idx < j; idx++){
        num_list[idx] = k;
    }
}
console.log(num_list.join(' '));

Array(n)로 n크기의 배열을 선언하고 fill(0)으로 값을 0으로 초기화한다.

6번 공 바꾸기

문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 바구니에는 공이 1개씩 들어있고, 처음에는 바구니에 적혀있는 번호와 같은 번호가 적힌 공이 들어있다. 

도현이는 앞으로 M번 공을 바꾸려고 한다. 도현이는 공을 바꿀 바구니 2개를 선택하고, 두 바구니에 들어있는 공을 서로 교환한다. 

공을 어떻게 바꿀지가 주어졌을 때, M번 공을 바꾼 이후에 각 바구니에 어떤 공이 들어있는지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다. 

둘째 줄부터 M개의 줄에 걸쳐서 공을 교환할 방법이 주어진다. 각 방법은 두 정수 i j로 이루어져 있으며, i번 바구니와 j번 바구니에 들어있는 공을 교환한다는 뜻이다. (1 ≤ i ≤ j ≤ N) 

도현이는 입력으로 주어진 순서대로 공을 교환한다.

출력

1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다.

N 사이즈의 배열에 요소들은 index+1 값으로 초기화한 상태로 시작된다.

C++

#include <iostream>
#include <vector>
#include <numeric>
#include <algorithm>
using namespace std;
int main(){
    int n, m, i, j;
    cin >> n >> m;
    vector<int> num_vec(n);
    iota(num_vec.begin(), num_vec.end(), 1);
    for (int k = 0; k < m; k++){
        cin >> i >> j;
        swap(num_vec[i - 1], num_vec[j - 1]);
    }
    for (int num : num_vec){
        cout << num << " ";
    }
    return 0;
}

iota 함수를 사용해서 컨테이너의 각 요소를 인덱스  + 1의 값으로 초기화한다.

algorithm의 swap 함수를 사용해서 컨테이너의 두 인덱스의 값을 교환한다.

C#

using System;
using System.Collections.Generic;
using System.Linq;
class Program{
    static void Main(string[] args){
        int[] input = Console.ReadLine().Split().Select(int.Parse).ToArray();
        int n = input[0];
        int m = input[1];
        List<int> num_list = new List<int>(Enumerable.Range(1, n));
        
        for (int l = 0; l < m; l++){
            int[] data = Console.ReadLine().Split().Select(int.Parse).ToArray();
            int i = data[0] - 1;
            int j = data[1] - 1;
            int tmp = num_list[i];
            num_list[i] = num_list[j];
            num_list[j] = tmp;
        }
        foreach(int num in num_list){
            Console.Write(num + " ");
        }
    }
}

new List <int>(Enumerable.Range(1, n)) 으로 각 요소를 1부터 n으로 초기화한다.

Python

input_str = list(map(int, input().split()))
n = input_str[0]
m = input_str[1]
num_list = list(range(1, n + 1))
for _ in range(m):
    i, j = map(int, input().split())
    num_list[i - 1], num_list[j - 1] = num_list[j - 1], num_list[i - 1]
print(" ".join(map(str, num_list)))

파이썬에서 리스트를 초기화할 때 range(1, n + 1) 초기화 시 값을 인덱스 + 1로 값을 넣는다.

출력 시 현재 리스트는 정수타입이므로 문자열로 변환 후 값마다 띄어쓰기를 추가해서 출력한다.

Node.js

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin', 'utf8').split('\n');
const [n, m] = input[0].split(' ').map(Number);
let num_list = Array.from({ length : n }, (_, i) => i + 1);
for (let k = 1; k <= m; k++){
    const [i, j] = input[k].split(' ').map(Number);
    [num_list[i - 1], num_list[j - 1]] = [num_list[j - 1], num_list[i - 1]];
}
console.log(num_list.join(' '));

Array.from({length : n}, (_, i) => i+1) 이렇게 인덱스 + 1로 각 요소들의 값을 초기화할 수 있다.

값을 교환하는 코드는 파이썬과 동일한 문법으로 우선 좌항과 우항의 오른쪽에 위치한 배열들의 값이 평가되면서 추출된다. 그리고 추출된 값들이 왼쪽에 대응하는 변수에 할당되면서 요소의 교환이 이루어진다. 

3번 최소, 최대

문제

N개의 정수가 주어진다. 이때, 최솟값과 최댓값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정수의 개수 N (1 ≤ N ≤ 1,000,000)이 주어진다. 둘째 줄에는 N개의 정수를 공백으로 구분해서 주어진다. 모든 정수는 -1,000,000보다 크거나 같고, 1,000,000보다 작거나 같은 정수이다.

출력

첫째 줄에 주어진 정수 N개의 최솟값과 최댓값을 공백으로 구분해 출력한다.

C++

vector를 사용해서 풀어본다.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
    int n;
    cin >> n;
    vector<int> vec(n);
    for (int i = 0; i < n; i++){
        cin >> vec[i];
    }
    vector<int>::iterator min_iter = min_element(vec.begin(), vec.end());
    vector<int>::iterator max_iter = max_element(vec.begin(), vec.end());
    cout << *min_iter << " " << *max_iter;
    return 0;
}

vector는 가변 배열이지만 사이즈를 특정해서 고정 크기로 사용할 수 있다.

algorithm을 사용해서 배열 안에서 min, max 요소를 찾아서 반복자를 반환받아서 값을 출력한다.

min, max_element 함수가 반복자를 반환하는 이유는 중복된 값을 저장할 필요 없이 컨테이너의 요소를 직접 참조할 수 있으며 컨테이너의 종류에 상관없이 유연하게 사용할 수 있기 때문이다. 또한 반복자에 접근해서 컨테이너의 요소를 직접 조작할 수도 있다.

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int main(){
    int n;
    cin >> n;
    vector<int> vec;
    for (int i = 0; i < n; i++){
        int val;
        cin >> val;
        vec.push_back(val);
    }
    vector<int>::iterator min_iter = min_element(vec.begin(), vec.end());
    vector<int>::iterator max_iter = max_element(vec.begin(), vec.end());
    cout << *min_iter << " " << *max_iter;
    return 0;
}

가변적으로 사용할 때는 push_back을 사용하면 된다.

C#

using System;
using System.Collections.Generic;
using System.Linq;
class Program{
    static void Main(string[] args){
        int n = int.Parse(Console.ReadLine());
        string[] input_arr = Console.ReadLine().Split(' ');
        List<int> num_list = new List<int>();
        for (int i = 0; i < n; i++){
            num_list.Add(int.Parse(input_arr[i]));
        }
        int min = num_list.Min();
        int max = num_list.Max();
        Console.WriteLine($"{min} {max}");
    }
}

C#에서는 List <T>를 사용할 수 있다. List는 Collections.Generic에 포함되어 있고 Min, Max를 사용하기 위해서 Linq도 사용한다.

리스트에 값을 넣는 과정도 Select와 ToList를 사용하면 한 번에 초기화할 수 있다.

string[] input_arr = Console.ReadLine().Split(' ');
List<int> num_list = new List<int>();
for (int i = 0; i < n; i++){
    num_list.Add(int.Parse(input_arr[i]));
}

//================================================
List<int> numbers = Console.ReadLine().Split(' ').Select(int.Parse).ToList();

Linq는 이외에도 배열이나 컬렉션 등을 처리할 때 유용한 기능들을 제공하기 때문에 함께 사용하는 경우가 많다. 하지만 강력한 기능인만큼 많은 메모리를 사용하기 때문에 주의해야 한다.

Python

n = int(input())
num_list = list(map(int, input().split()))
min_val = min(num_list)
max_val = max(num_list)
print(f"{min_val} {max_val}")

파이썬은 list()를 사용할 수 있다.

map은 해당 함수가 반복 가능한 객체의 각 요소에 대해 적용된 결과를 반환하는 map객체를 생성하는데 이 객체는 필요에 따라 리스트나 튜플로 변환하여 사용할 수 있다.

map을 사용해서 입력되는 값을 공백으로 구분하고 각 문자열을 정수로 변환하여 반환되는 map으로 list를 초기화한다.

Node.js

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin','utf8').split('\n');
const n = parseInt(input[0]);
const nums = input[1].split(' ').map(Number);
let min_val = nums[0];
let max_val = nums[0];
for (int i = 1; i < n; i++){
    if (nums[i] < min_val)
        min_val = nums[i]
    else if (nums[i] > max_val)
        max_val = nums[i]
}
console.log(`${min_val} ${max_val}`);

js도 map을 사용해서 배열을 초기화시킬 수 있다. 입력받은 값을 Number함수로 각 요소를 숫자로 변환한 map을 반환한다.

최대 최소를 구하는 방법은 함수를 쓰지 않고 각 요소들의 크기를 작으면 min, 크면 max로 비교하여 구할 수 있다.

js에서도 최대 최솟값을 구하는 Math.min, Math.max를 지원한다.

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin','utf8').split('\n');
const n = parseInt(input[0]);
const nums = input[1].split(' ').map(Number);
const min_val = Math.min(...nums);
const max_val = Math.max(...nums);
console.log(`${min_val} ${max_val}`);

1번 개수 세기

문제 

총 N개의 정수가 주어졌을 때, 정수 v가 몇 개인지 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정수의 개수 N(1 ≤ N ≤ 100)이 주어진다. 둘째 줄에는 정수가 공백으로 구분되어 있다. 셋째 줄에는 찾으려고 하는 정수 v가 주어진다. 입력으로 주어지는 정수와 v는 -100보다 크거나 같으며, 100보다 작거나 같다.

출력

첫째 줄에 입력으로 주어진 N개의 정수 중에 v가 몇 개인지 출력한다.

배열에 저장하고 조건에 맞는 값을 출력한다.

C++

#include <iostream>
using namespace std;
int main(){
    int n, v, count;
    cin >> n;
    int arr[n];
    for (int i = 0; i < n; i++){
        cin >> arr[i];
    }
    
    cin >> v;
    for (int i = 0; i < n; i++){
        if (arr[i] == v)
            count++;
    }
    cout << count;
    return 0;
}

입력될 크기인 n 만큼의 크기의 배열을 선언하고 반복문으로 배열에 값들을 저장한다.

그리고 v의 입력을 받아 반복문으로 배열을 순회하면서 같은 값인 경우 count를 증가시키고 최종 개수를 출력한다.

C#

using System;
class Program{
    static void Main(string[] args){
        int n = int.Parse(Console.ReadLine());
        int[] arr = new int[n];
        string input = Console.ReadLine().Trim();
        string[] input_arr = input.Split(' ');
        for (int i = 0; i < n; i++){
            arr[i] = int.Parse(input_arr[i]);
        }
        int v = int.Parse(Console.ReadLine());
        int count = 0;
        for (int i = 0; i < n; i++){
            if (arr[i] == v)
                count++;
        }
        Console.WriteLine(count);
    }
}

Python

n = int(input())
values = input().strip().split(' ')
arr = [0] * n
for i in range(n):
    arr[i] = int(values[i]);
v = int(input())
count = 0;
for i in range(n):
    if arr[i] == v:
        count += 1
print(count)

Node.js

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin','utf8').split('\n');
const n = parseInt(input[0]);
const values = input[1].split(' ');
const v = parseInt(input[2]);
let count = 0;
const val_arr = new Array(n);
for (let i = 0; i < n; i++){
    val_arr[i] = parseInt(values[i]);
}
for (let i = 0; i < n; i++){
    if (val_arr[i] === v)
        count++;
}
console.log(count);

더 효율적인 방법도 있지만 1차원 배열을 사용해서 푸는걸 중점으로 두었기 때문에 불필요한 변수와 반복문이 사용되기도 한다.

2번 X보다 작은 수

문제

정수 N개로 이루어진 수열 A와 정수 X가 주어진다. 이때, A에서 X보다 작은 수를 모두 출력하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 X가 주어진다. 

(1 ≤ N, X ≤ 10,000) 둘째 줄에 수열 A를 이루는 정수 N개가 주어진다. 주어지는 정수는 모두 1보다 크거나 같고, 10,000보다 작거나 같은 정수이다.

출력

X보다 작은 수를 입력받은 순서대로 공백으로 구분해 출력한다. X보다 작은 수는 적어도 하나 존재한다.

1번을 풀었던 것처럼 배열을 사용하는데 중점을 둔다.

C++

#include <iostream>
using namespace std;
int main(){
    cin.tie(NULL);
    ios_base::sync_with_stdio(false);
    int n, x;
    cin >> n;
    cin >> x;
    int arr[n];
    for (int i = 0; i < n; i++){
        cin >> arr[i];
    }
    for (int a : arr){
        if (a < x){
            cout << a << " ";
        }
    }
    return 0;
}

배열의 모든 요소를 순회할 때는 for (int a : arr)를 사용할 수 있다.

C#

using System;
class Program{
    static void Main(string[] args){
        string input = Console.ReadLine();
        string[] input_arr = input.Split(' ');
        int n = int.Parse(input_arr[0]);
        int x = int.Parse(input_arr[1]);
        input = Console.ReadLine();
        input_arr = input.Split(' ');
        int[] arr = new int[n];
        for (int i = 0; i < n; i++){
            arr[i] = int.Parse(input_arr[i]);
        }
        
        foreach(int a in arr){
            if (a < x){
                Console.Write($"{a} ");
            }
        }
    }
}

Python

n, x = map(int, input().split())
str_arr = input().split();
arr = [0] * n;
for i in range(n):
    arr[i] = int(str_arr[i])
for i in range(n):
    if arr[i] < x :
        print(f"{arr[i]} ", end="")

출력 시 print는 기본적으로 줄 바꿈을 실행하므로 end=""로 설정해서 줄 바꿈을 제거한다.

Node.js

const fs = require('fs');
const input = fs.readFileSync('/dev/stdin','utf8').split('\n');
const [n, x] = input[0].split(' ').map(Number);
const values = input[1].split(' ').map(Number);
let result = [];
values.forEach(value => {
    if (value < x) {
        result.push(value);
    }
});

console.log(result.join(' '));

각각의 입력 값들을 정리하고 결과를 따로 배열에 저장한 뒤 일괄적으로 출력한다.

1, 2번 문제들은 고정 크기의 배열을 사용해서 풀었지만 가변 배열인 vector 나 list 등을 사용하면 더 간단하게 처리할 수 있는 문제이기 때문에 이후에는 가변 배열을 사용해서 풀어보기로 한다.