Z 字形变换

Leetcode题目链接

描述

详见中文题目链接。

解答

class Solution {
public:
    string convert(string s, int numRows) {
        if (numRows < 2) return s;

        vector<string> rows(numRows);
        int i = 0, flag = -1;

        for (char ch : s) {
            rows[i].push_back(ch);
            if (i == 0 || i == numRows - 1) {
                flag = -flag;
            }
            i += flag;
        }

        string res;
        for (const string& row : rows) {
            res += row;
        }


        return res;
    }
};