[수학] 5. 문제 풀이 (2)

in 자료구조 on 수학




1. 문자열 내 마음대로 정렬하기😭

programmers

문자열로 구성된 리스트 strings와, 정수 n이 주어졌을 때, 각 문자열의 인덱스 n번째 글자를 기준으로 오름차순 정렬하려 합니다.
예를 들어 strings가 [“sun”, “bed”, “car”]이고 n이 1이면 각 단어의 인덱스 1의 문자 “u”, “e”, “a”로 strings를 정렬합니다.

strings는 길이 1 이상, 50이하인 배열입니다.
strings의 원소는 소문자 알파벳으로 이루어져 있습니다.
strings의 원소는 길이 1 이상, 100이하인 문자열입니다.
모든 strings의 원소의 길이는 n보다 큽니다.
인덱스 1의 문자가 같은 문자열이 여럿 일 경우, 사전순으로 앞선 문자열이 앞쪽에 위치합니다.


function solution(strings, n) {
  // n 위치의 문자로 정렬

  // n 위치의 문자가 같ㅇ르 경우, 사전순으로 정렬한다.

  return strings.sort((x, y) => x[n] == y[n] ? (x > y ? 1 : -1) : (x[n] > y[n] ? 1 : -1))
}

sort 콜백함수 더 공부하기..



2. 직사각형 별찍기

이 문제에는 표준 입력으로 두 개의 정수 n과 m이 주어집니다.
별(*) 문자를 이용해 가로의 길이가 n, 세로의 길이가 m인 직사각형 형태를 출력해보세요.

process.stdin.setEncoding('utf8');
process.stdin.on('data', data => {
    const n = data.split(" ");
    const a = Number(n[0]), b = Number(n[1]);

    for (let i = 0; i< b; i++){
        let star = ""
        for (let j = 0; j<a; j++){
            star += "*"
         }
        console.log(star)
}
});




const row = '*'.repeat(a)
    for(let i =0; i < b; i++){
        console.log(row)
    }



3. 두 정수 사이의 합

programmers

두 정수 a, b가 주어졌을 때 a와 b 사이에 속한 모든 정수의 합을 리턴하는 함수, solution을 완성하세요.
예를 들어 a = 3, b = 5인 경우, 3 + 4 + 5 = 12이므로 12를 리턴합니다.

function solution(a, b) {
    if (a > b){
        [a, b] = [b, a]
        console.log(a, b)
    }
    let sum = 0
    for (let i = a; i <= b; i++){
        sum += i
    }
    return sum;
}



// 가우스
return (a+b)*(Math.abs(b-a)+1)/2;
  • 시간복잡도 O(n) > O(1)


for (var i = Math.min(a, b); i <= Math.max(a, b); i++) s += i;
  return s;



4. 부족한 금액 계산하기

programmers

새로 생긴 놀이기구는 인기가 매우 많아 줄이 끊이질 않습니다. 이 놀이기구의 원래 이용료는 price원 인데, 놀이기구를 N 번 째 이용한다면 원래 이용료의 N배를 받기로 하였습니다.
즉, 처음 이용료가 100이었다면 2번째에는 200, 3번째에는 300으로 요금이 인상됩니다.
놀이기구를 count번 타게 되면 현재 자신이 가지고 있는 금액에서 얼마가 모자라는지를 return 하도록 solution 함수를 완성하세요.
단, 금액이 부족하지 않으면 0을 return 하세요.

function solution(price, money, count) {
    let charge = 0
    for (let i = 1; i <= count; i++){
        charge += price * i
    }

    return money-charge > 0 ?  0 : charge-money
}



// 가우스...!
// count 횟수 먼저 다 구한다음에 한번 곱하기
function solution(price, money, count) {
    const tmp = price * count * (count + 1) / 2 - money;
    return tmp > 0 ? tmp : 0;
}



5. K번째 수

programmers
배열 array의 i번째 숫자부터 j번째 숫자까지 자르고 정렬했을 때, k번째에 있는 수를 구하려 을 때 나온 결과를 배열에 담아 return 하도록 solution 함수를 작성해주세요.

function solution(array, commands) {
    let result = []
    for (let i =0; i< commands.length; i++){
        let arr = array.slice(commands[i][0]-1,commands[i][1])
        arr.sort((x,y) => x-y)
        result.push(arr[commands[i][2]-1])
    }
    return result
}



// map 사용하기
function solution(array, commands) {
    return commands.map(commands => {
      const [s, e, p] = command
        return array.slice(s-1,e).sort((a, b) => a - b)[p-1]
    });
}



6. 나누어 떨어지는 숫자 배열

programmers
array의 각 element 중 divisor로 나누어 떨어지는 값을 오름차순으로 정렬한 배열을 반환하는 함수, solution을 작성해주세요.
divisor로 나누어 떨어지는 element가 하나도 없다면 배열에 -1을 담아 반환하세요.

function solution(arr, divisor) {
  let result = [];
  for (let i = 0; i < arr.length; i++) {
    if (arr[i] % divisor === 0) {
      result.push(arr[i]);
    }
  }
  if (result.length === 0 ) return [-1];
  result.sort((x, y) => x - y);
  return result;
}

// filter 메소드 사용
function solution(arr, divisor) {
    var answer = arr.filter(v => v%divisor == 0);
    return answer.length == 0 ? [-1] : answer.sort((a,b) => a-b);
}



7. 같은 숫자는 싫어

programmers

배열 arr가 주어집니다. 배열 arr의 각 원소는 숫자 0부터 9까지로 이루어져 있습니다. 이때, 배열 arr에서 연속적으로 나타나는 숫자는 하나만 남기고 전부 제거하려고 합니다.
단, 제거된 후 남은 수들을 반환할 때는 배열 arr의 원소들의 순서를 유지해야 합니다. 예를 들면,

arr = [1, 1, 3, 3, 0, 1, 1] 이면 [1, 3, 0, 1] 을 return 합니다.
arr = [4, 4, 4, 3, 3] 이면 [4, 3] 을 return 합니다.
배열 arr에서 연속적으로 나타나는 숫자는 제거하고 남은 수들을 return 하는 solution 함수를 완성해 주세요.

function solution(arr)
{
  let result = [];
  result.push(arr[0]);
  for (let i = 1; i < arr.length; i++) {
    if (arr[i] != arr[i - 1]) result.push(arr[i]);
  }
  return result;
}

function solution(arr)
{
    return arr.filter((val,index) => val != arr[index+1]);
}