프로그래밍 문제/c++ 문제
c++ 문제 44. 배열의 요소들을 왼쪽 또는 오른쪽으로 밀기
지나팩
2023. 12. 29. 18:07
방향과 이동 횟수를 입력받아 주어진 배열의 요소들을 밀어주는 코드를 작성하세요.
출력 예)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
#include <iostream>
#include <cmath> // sqrt와 pow 함수 사용을 위한 헤더
void rotateArray(int* arr, int n, int k, int direction) {
if (direction == 0) return; // 방향이 0이면 순환하지 않음
int* temp = new int[n];
k = k % n; // k가 배열 길이보다 큰 경우를 대비하여 나머지 연산 사용
if (direction > 0) { // 오른쪽으로 순환
for (int i = 0; i < n; i++) {
temp[(i + k) % n] = arr[i];
}
}
else { // 왼쪽으로 순환
for (int i = 0; i < n; i++) {
temp[(i - k + n) % n] = arr[i];
}
}
for (int i = 0; i < n; i++) {
arr[i] = temp[i];
}
delete[] temp;
}
int main()
{
int arr[] = { 1, 2, 3, 4, 5, 6, 7 };
int n = sizeof(arr) / sizeof(arr[0]);
int k = 3; // 오른쪽으로 3단계 이동
int dir = 0;
cout << "Original array: ";
for (int i = 0; i < n; i++) {
cout << arr[i] << " ";
}
cout << endl;
cout << "순환 방향을 설정하세요: 음수(왼쪽), 양수(오른쪽) : ";
cin >> dir;
cout << "이동 횟수를 설정하세요: ";
cin >> k;
cout << endl;
rotateArray(arr, n, k, dir);
for (int i = 0; i < n; i++) {
cout << arr[i] << " ";
}
return 0;
}
|
cs |