SOJ 1036. Crypto Columns

题目大意：一个字符串，去掉空格和符号后变成"plaintext"。另一个字符串叫"keyword"。将plaintext排成多行，每行有keyword.size的宽度，不能排满一列的用别的字符填满，生成了字符矩阵。每次选取keyword中字典序最小的字符对应的列，作为需要选取的矩阵的列，被选择过的字符不再被选择，最后形成一行新的字符串，作为输出字符串。
解题思路：对keyword中的字符及其对应的列进行排序，利用排序结果还原成字符矩阵，然后按行输出字符矩阵。
代码如下：

#include <iostream>
#include <string>
#include <vector>
using namespace std;

const int COLS = 15;
const int ROWS = 105;

string kw;
string ct;

char m[ROWS][COLS];

void init() {
    for (int i = 0; i < ROWS; i++) {
        for (int j = 0; j < COLS; j++) {
            m[i][j] = 0;
        }
    }
}

struct Elem {
    char ch;
    int col;
};

Elem elems[COLS];

bool cmp(const Elem & e1, const Elem &e2) {
    /*if (e1.ch > e2.ch) {
        return true;
    } else if (e1.ch < e2.ch) {
        return false;
    } else {
        if (e1.col > e2.col) {
            return true;
        } else if (e1.col < e2.col){
            return false;
        } else {
            return false;
        }
    }*/

    return (e1.ch > e2.ch) || (e1.ch == e2.ch && e1.col > e2.col);
}

int main() {
    string end = "THEEND";

    while (cin >> kw, kw != end) {
        cin >> ct;
        int c = kw.size();
        int r = ct.size() / kw.size();

        for (int i = 0; i < c; i++) {
            elems[i].ch = kw[i];
            elems[i].col = i;
        }

        for (int i = 0; i < c; i++) {
            for (int j = 0; j < c - i - 1; j++) {
                if (cmp(elems[j], elems[j + 1])) {
                    Elem elem = elems[j];
                    elems[j] = elems[j+1];
                    elems[j+1] = elem;
                }
            }
        }

        for (int i = 0; i < c; i++) {
            int begin = r * i;
            //int end = begin + r;
            int col = elems[i].col;

            for (int j = 0; j < r; j++) {
                m[j][col] = ct[begin + j];
            }
        }

        for (int i = 0; i < r; i++) {
            for (int j = 0; j < c; j++) {
                cout << m[i][j];
            }
        }

        cout << endl;
    }

    return 0;
}