按给定 X 的元素对数组进行分区

Question

我正在尝试查找 数组的分区，通过检查变量 x 的条件，当小于 x 时，它们将位于一侧或另一侧。 但我的代码需要一些更正。 这里我找不到错误，如果你能帮助我，我会很感激你。 代码是：-

#include<iostream>
using namespace std;
int partition(int arr[],int n,int x){
    for(int i=0;i<n;){
        if(arr[i]<x){
            i++;
        }
        else if(arr[i]==x){   
            int temp=arr[i];
            arr[i]=arr[n];
            arr[n]=temp;
            i--;
        }
        else if(arr[i]>x){
            int temp=arr[i];
            for(int j=i;j<n;j++){
                arr[j]=arr[j+1];   
            }
            arr[n]=temp;
           i--;
        }
    } 
    return 0;
}


int main(){
    int n;
    cin>>n;
    int arr[n];
    for(int i=0;i<n;i++){
        cin>>arr[i];    
    }
    int x;
    cin>>x;
    
    partition(arr,n,x);
    
     for(int i=0;i<n;i++){
        cout<<arr[i]<<"\t";
     }
    return 0;
}

Input >> array={2,10,15,1,3,15} ,x=10

Expected << {2,1,3,10,15,15}

Output I get << nothing .

Answer 1

代码没有给出任何输出，因为首先，“cin”和“cout”是大写的，这在语法上是不正确的，其次，变量 j 在循环语句中的大小写不同，而在第二个 else 中的主体分区函数中的 -if 子句，main() 函数中第一个 for 循环中的“I”也是如此。解决这个问题，你应该很高兴。

Answer 2

首先在 C++ 中，数组的大小必须是编译时常量。因此，例如，考虑以下示例：

int n = 10;
int arr[n]; //INCORRECT

上面的正确写法是：

const int n = 10;
int arr[n]; //CORRECT

同样，在您的代码中，

int n;
cin>>n;
int arr[n]; //INCORRECT because n is not a constant expression

第二，在你的代码中，当你写的时候：

arr[n] = temp; Undefined behavior

你会越界，所以你有未定义的行为。

解决方案

您可以使用std::stable_partition 和std::vector 来解决您的问题，如下所示：

#include <iostream>
#include <vector>
#include <algorithm>
int main()
{
    int n;
    std::cout <<"Enter n:"<<std::endl;
    std::cin >> n;
    
    std::vector<int> arr(n); //create a vector of size n instead of an array 
    
    std::cout<<"Enter elements: "<<std::endl;
    //iterate and take input from user
    for(int &elem: arr){
        std::cin >> elem ;     
    }
    int x;
    std::cout << "Enter x: "<<std::endl;
    std::cin>>x;
    
    //use std::partition
    std::stable_partition(arr.begin(), arr.end(), [x](int i){return (i < x);});
    
    std::cout<<"This is the partitioned vector: "<<std::endl;
    
    for(int i=0;i<n;i++)
    {
        std::cout<<arr[i]<<"\t";
    }
    return 0;
}

输出

上述程序的输出如下：

Enter n:
6
Enter elements: 
2
10
15
1
3
15
Enter x: 
10
This is the partitioned vector: 
2       1       3       10      15      15

可以看到here。