C++: как отсортировать сложную матрицу на основе вещественных частей сложного вектора

У меня есть сложный вектор (VSII_Complex, содержащий собственные значения) и комплексная матрица (CUII_Complex, содержащая собственные векторы). Каждый элемент VSII_Complex соответствует столбцу CUII_Complex. Моя проблема в том, что я хочу отсортировать собственные значения внутри VSII_Complex на основе их реальной части (НЕ мнимой части), и мне придется сортировать столбцы CUII_Complex в соответствии с отсортированным VSII_Complex. Следующий код разработан моим другом, но мне кажется, что с этим кодом что-то не так, но я не могу в этом разобраться. Интересно, может ли кто-нибудь сказать, что не так, если таковые имеются.

EIG eigA=EIG(m_IIStiffnessAct,m_IIMassAct,true);
    ComplexColumnVector VSII_Complex=eigA.eigenvalues();
    ComplexMatrix CUII_Complex=eigA.eigenvectors();
///// make eigenvalues in decreasing order, so do eigenvectors
    for (long ii = 0; ii < VSII_Complex.rows(); ii++)
    {
        for (long jj = ii+1; jj < VSII_Complex.rows(); jj++)
        {
          if (VSII_Complex(ii).real() < VSII_Complex(jj).real())
          {
              Complex temp = VSII_Complex(ii);
              VSII_Complex(ii) = VSII_Complex(jj);
              VSII_Complex(jj) = temp;             
              for (long kk = 0; kk < CUII_Complex.rows(); kk++)
              {
                  Complex tempVec =  CUII_Complex(kk,ii);
                  CUII_Complex(kk,ii) = CUII_Complex(kk,jj);
                  CUII_Complex(kk,jj) = tempVec;
              }
          }
        }
    }

c++ sorting complex-numbers eigenvalue eigenvector
person user3405291    schedule 03.11.2014    source источник

Ответы (2)


arrow_upward
1
arrow_downward

Просто используйте встроенную сортировку, которая возвращает позицию, в которой элемент находился ранее.

//couldn't find this in the docs, I'm overlooking something probably:
void swapColumns (ComplexMatrix &mat, octave_idx_type colA, octave_idx_type colB)
{
    if(colA == colB) return;
    ComplexColumnVector temp = mat.column(colA);
    mat.insert(mat.column(colB),0,colA);
    mat.insert(temp,0,colB);
}


bool isRealGreater(const Complex& a, const Complex& b)
{
    return a.real() > b.real();
}

//presumably in another function
//int func() {
    EIG eigA=EIG(m_IIStiffnessAct,m_IIMassAct,true);

    ComplexColumnVector VSII_Complex=eigA.eigenvalues();
    ComplexMatrix CUII_Complex=eigA.eigenvectors();
///// make eigenvalues in decreasing order, so do eigenvectors

    //create indices from 1-len(VSII_Complex)
    Array<octave_idx_type> sort_order(VSII_Complex.length(),0);
    for(int i =0 ; i< sort_order.length(); i++)
    {sort_order.elem(i)= i;}

    //create sorting object and sort VSII_Complex in descending order of the real component
    octave_sort<Complex> sort_obj(&isRealGreater);
    sort_obj.sort(VSII_Complex.jit_slice_data(), sort_order.jit_slice_data(), VSII_Complex.length());

    //swap the columns of CUII_Complex in the same way VSII_Complex got sorted
    for(octave_idx_type i=0; i<sort_order.length(); i++)
    {
        if(sort_order.elem(i) > i)
        {
            swapColumns(CUII_Complex,i,sort_order.elem(i));
        }
    }

//}

На самом деле не проверял это, потому что мне было лень устанавливать октаву, я просто прочитал документы.

person PeterT    schedule 03.11.2014
comment
Я добавил код C++ в качестве решения этого вопроса в своем ответе ниже. Но ваше решение мне кажется лучше. - person user3405291; 03.11.2014

arrow_upward
0
arrow_downward

Я решил проблему с помощью следующего кода, я разработал этот код на основе алгоритма на следующей веб-странице. Но ответ PeterT мне кажется лучше.

http://www.learncpp.com/cpp-tutorial/64-sorting-an-array-using-selection-sort/

///// make eigenvalues in decreasing order, so do eigenvectors
    ComplexColumnVector ComplexColumnVector_toBesSorted=VSII_Complex;
    ComplexMatrix ComplexMatrix_toBeSorted=CUII_Complex;
    for (long idx_start = 0; idx_start < ComplexColumnVector_toBesSorted.rows(); idx_start++)
    {
        long idx_smallest=idx_start;
        for (long idx_current = idx_start+1; idx_current < ComplexColumnVector_toBesSorted.rows(); idx_current++)
        {
          if (ComplexColumnVector_toBesSorted(idx_current).real() < ComplexColumnVector_toBesSorted(idx_smallest).real())
              idx_smallest=idx_current;
        }
        Complex Complex_temp=ComplexColumnVector_toBesSorted(idx_start);
        ComplexColumnVector_toBesSorted(idx_start)=ComplexColumnVector_toBesSorted(idx_smallest);
        ComplexColumnVector_toBesSorted(idx_smallest)=Complex_temp;
        for (long kk = 0; kk < ComplexMatrix_toBeSorted.rows(); kk++)
        {
            Complex Complex_temp_2 =  ComplexMatrix_toBeSorted(kk,idx_start);
            ComplexMatrix_toBeSorted(kk,idx_start) = ComplexMatrix_toBeSorted(kk,idx_smallest);
            ComplexMatrix_toBeSorted(kk,idx_smallest) = Complex_temp_2;
        }
    }
person user3405291    schedule 03.11.2014