使用c ++

时间:2016-05-19 21:15:08

标签: c++ multithreading sorting synchronization quicksort

我使用quicksort方法实现multi-thread程序,在C ++中使用Portfolio任务。

  

组合任务的方法是维护任务队列。每   自由线程从投资组合中挑选任务,必要时执行它   生成新的子任务并将其放入投资组合

但我不确定什么是对的!在我看来,在一个thread中,算法比两个或四个thread工作得更快。我能以某种方式搞砸同步吗?

感谢任何人帮助我。

代码:

#include <thread>
#include <chrono>
#include <mutex>
#include <condition_variable>
#include <iostream>
#include <queue>
#include <vector>
#include <set>
#include <ctime>
#include <algorithm>

using namespace std;

//print array
template <typename T>
void print(const vector<T> &arr)
{
    for (size_t i = 0; i < arr.size(); i++)
        cout << arr[i] << " ";
    cout << endl;
}

//queue tasks
queue< pair<int, int> > tasks;
//mutexs for set and queue task
mutex q_mutex, s_mutex;
//condition variable
condition_variable cv;
//set
set<int> ss;

//partition algorithm
template <typename T>
int partition(vector<T> &arr, int l, int r)
{
    T tmp = arr[r]; //as pivot element
    int i = l - 1;

    for (int j = l; j <= r - 1; j++)
        if (arr[j] < tmp)
        {
            i++;
            swap(arr[i], arr[j]);       
        }

    swap(arr[i + 1], arr[r]);
    i++;
    return i;
}

//quick sort
template <typename T>
void quick_sort(vector<T> &arr)
{
    while (true)
    {
        unique_lock<mutex> u_lock(q_mutex); //lock mutex

        //sort is fineshed
        if ( ss.size() == arr.size() ) //u_lock.unlock()
            return;

        //if queue task is not empty 
        if ( tasks.size() > 0 )
        {
            //get task from queue
            pair<int, int> cur_task = tasks.front();            
            tasks.pop();

            int l = cur_task.first, r = cur_task.second;        

            if (l < r)
            {
                int q = partition(arr, l, r); //split array

                //Add indexes in set
                s_mutex.lock();
                ss.insert(q);
                ss.insert(l);
                ss.insert(r);
                s_mutex.unlock();

                //push new tasks for left and right part
                tasks.push( make_pair(l, q - 1) );
                tasks.push( make_pair(q + 1, r) );

                //wakeup some thread which waiting 
                cv.notify_one();
            }
        }
        else
            //if queue is empty
            cv.wait(u_lock);
    }
}

//Size array
const int ARR_SIZE = 100000;
//Count threads
const int THREAD_COUNT = 8;

thread thrs[THREAD_COUNT];

//generatin array
void generate_arr(vector<int> &arr)
{
    srand(time( NULL ));

    std::generate(arr.begin(), arr.end(), [](){return rand() % 10000; });
}

//check for sorting
bool is_sorted(const vector<int> &arr)
{
    for (size_t i = 0; i < arr.size() - 1; i++)
        if ( ! (arr[i] <= arr[i + 1]) ) 
            return false;
    return true;
}

int main()
{
    //time
    clock_t start, finish;

    vector<int> arr(ARR_SIZE);

    //generate array
    generate_arr(arr);

    cout << endl << "Generating finished!" << endl << endl;

    cout << "Array before sorting" << endl << endl;

    //Before sorting
    print(arr);

    cout << endl << endl;

    cout << "Checking is_sorted finished! The result is " << (is_sorted(arr) == 0? "false": "true") << "." << endl << endl;

    //add task
    tasks.push( make_pair(0, arr.size() - 1) );

    //==================================================
    start = clock();

    for (int i = 0; i < THREAD_COUNT; i++)
        thrs[i] = thread( quick_sort<int>, ref(arr) );

    finish = clock();
    //==================================================

    for (auto& th : thrs)
        th.join();

    cout << "Sorting finished!" << endl << endl;

    cout << "Array after sorting" << endl << endl;
    //After sorting
    print(arr);

    cout << endl << endl;

    cout << "Checking is_sorted finished! The result is " << (is_sorted(arr) == 0? "false": "true") << "." << endl << endl;

    cout << "Runtime: " << (double)(finish - start) / CLOCKS_PER_SEC << endl;

    return 0;
}

2 个答案:

答案 0 :(得分:6)

有很多因素可以解决问题所涉及的线程数量。其中,

  • 您需要具有实际的并发性,而不仅仅是多个线程。正如@ Rakete1111和@ user1034749都观察到的那样,你不会。

  • 标准quicksort具有良好的引用局部性,特别是当分区大小变小时,但是由于给定数组元素的责任很可能在每次分区时交换到不同的线程,因此您的技术会抛弃很多。

  • 此外,互斥操作并不是特别便宜,而且当分区变小时,相对于实际排序量,你会开始做很多这些操作。

  • 使用比物理内核更多的线程是没有意义的。四个线程可能不是太多,但这取决于你的硬件。

以下是一些可以提高多线程性能的方法:

  1. 在方法quick_sort()中,不要在实际排序过程中锁定互斥锁q_mutex,就像您目前所做的那样(您使用的unique_lock构造函数锁定了互斥锁,而您请勿在{{1​​}})的生命周期内解锁。

  2. 切换到小于某个阈值大小的分区的普通递归技术。您必须进行测试才能找到一个好的特定阈值;也许它需要是可调的。

  3. 在每次分区时,让每个线程只将一个子分区发布到组合中;让它以递归的方式处理另一个 - 或者更好,迭代地处理。实际上,将它设为您发布的较小的子分区,因为这样可以更好地约束投资组合的大小。

  4. 您可能还会考虑增加运行测试的元素数量。 100000并不是那么多,你可能会看到更大问题的不同性能特征。在现代硬件上进行这样的测试,1000000个元素根本不合理。

答案 1 :(得分:1)

在我看来,您应该将投资组合任务的行为捕获到一个类中。

template <typename TASK, unsigned CONCURRENCY>
class Portfolio {
    std::array<std::thread, CONCURRENCY> workers_;
    std::deque<TASK> tasks_;
    std::mutex m_;
    std::condition_variable cv_;
    std::atomic<bool> quit_;
    void work () {
        while (!quit_) {
            TASK t = get();
            if (quit_) break;
            t();
        }
    }
    TASK get () {
        std::unique_lock<std::mutex> lock(m_);
        while (tasks_.empty()) {
            cv_.wait(lock);
            if (quit_) return TASK();
        }
        TASK t = tasks_.front();
        tasks_.pop_front();
        if (!tasks_.empty()) cv_.notify_one();
        return t;
    }
public:
    void put (TASK t) {
        std::unique_lock<std::mutex> lock(m_);
        tasks_.push_back(t);
        cv_.notify_one();
    }
    Portfolio ();
    ~Portfolio ();
};

构造函数将使用每个调用work()方法的线程初始化worker。析构函数将设置quit_,发出所有线程的信号,并加入它们。

然后,您的快速排序可以简化:

template <typename T, unsigned WORK_SIZE, typename PORTFOLIO>
QuickSortTask {
    std::reference_wrapper<PORTFOLIO> portfolio_;
    std::reference_wrapper<std::vector<T>> arr_;
    int start_;
    int end_;

    QuickSortTask (PORTFOLIO &p, std::vector<T> &a, int s, int e)
        : portfolio_(p), arr_(a), start_(s), end_(e)
        {}

    void operator () () {
        if ((end_ - start_) > WORK_SIZE) {
            int p = partition(arr_, start_, end_);
            portfolio_.put(QuickSortTask(portfolio_, arr_, start_, p-1));
            portfolio_.put(QuickSortTask(portfolio_, arr_, p+1, end_));
        } else {
            regular_quick_sort(arr_, start_, end_);
        }
    }
};

不幸的是,这种制定并行快速排序的方式不太可能产生大的加速。你想要做的是并行化分区任务,这需要至少一个单线程计算通过(涉及数据比较和交换),然后才能开始并行化。

首先将数组划分为WORK_SIZE子数组,并行执行快速排序,然后合并结果以创建排序向量可能会更快。