尝试使用堆栈顺时针旋转数组时出错
getting error while trying to rotate array clock wise using stack
给定一个大小为 N 的未排序数组 arr[],将其旋转 D 个元素(顺时针)。
输入
输入的第一行包含 T,表示测试用例的数量。每个测试用例的第一行包含两个 space 分隔的元素,N 表示数组的大小,一个整数 D 表示旋转的数字大小。后续行将是 N space 个分隔的数组元素
输出
对于每个测试用例,在新的一行中,输出旋转后的数组
例子
输入
1 2 3 4 5
输出
3 4 5 1 2
#include <iostream>
#include <stack>
using namespace std;
void rotate(int *a,int s,int r) {
stack<int> st;
for(int i=0;i<r;i++) {
st.push(a[i]);
}
for(int j=r;j<s;j++) {
a[j-r] = a[j];
}
for(int k=s-1;k>r+1;k--) {
a[k] = st.top();
st.pop();
}
for(int l=0;l<s;l++) {
cout<<a[l]<<" ";
}
}
int main() {
//code
int T;
cin>>T;
while(T--) {
int N,r;
cin>>N>>r;
int A[N];
for(int i=0;i<N;i++) {
cin>>A[i];
}
rotate(A,N,r);
cout<<endl;
}
return 0;
}
按照你的逻辑,你的回填部分好像有问题。
for(int k=s-1;k>=s-r;k--) { // change k>r+1 to k>=s-r
a[k] = st.top();
st.pop();
}
对不起我的错,旋转函数中的第三个for循环应该有k>s-r-1
给定一个大小为 N 的未排序数组 arr[],将其旋转 D 个元素(顺时针)。
输入 输入的第一行包含 T,表示测试用例的数量。每个测试用例的第一行包含两个 space 分隔的元素,N 表示数组的大小,一个整数 D 表示旋转的数字大小。后续行将是 N space 个分隔的数组元素
输出 对于每个测试用例,在新的一行中,输出旋转后的数组
例子 输入 1 2 3 4 5
输出 3 4 5 1 2
#include <iostream>
#include <stack>
using namespace std;
void rotate(int *a,int s,int r) {
stack<int> st;
for(int i=0;i<r;i++) {
st.push(a[i]);
}
for(int j=r;j<s;j++) {
a[j-r] = a[j];
}
for(int k=s-1;k>r+1;k--) {
a[k] = st.top();
st.pop();
}
for(int l=0;l<s;l++) {
cout<<a[l]<<" ";
}
}
int main() {
//code
int T;
cin>>T;
while(T--) {
int N,r;
cin>>N>>r;
int A[N];
for(int i=0;i<N;i++) {
cin>>A[i];
}
rotate(A,N,r);
cout<<endl;
}
return 0;
}
按照你的逻辑,你的回填部分好像有问题。
for(int k=s-1;k>=s-r;k--) { // change k>r+1 to k>=s-r
a[k] = st.top();
st.pop();
}
对不起我的错,旋转函数中的第三个for循环应该有k>s-r-1