[LeetCode] ZigZag Conversion

标签：c++   leetcode   

ZigZag Conversion

The string "PAYPALISHIRING" is written in a zigzag pattern on a given number of rows like this: (you may want to display this pattern in a fixed font for better legibility)

P   A   H   N
A P L S I I G
Y   I   R

Write the code that will take a string and make this conversion given a number of rows:

string convert(string text, int nRows);

解题思路：

1、一般这种矩形的变换，直接想到的办法是按照它的规则模拟一遍即可。下面是相关代码，其时间复杂度和空间复杂度均为O(n*nRows)

class Solution {
public:
    string convert(string s, int nRows) {
        if(nRows<2){
            return s;
        }
        int len=s.length();
        char matrix[nRows][len];
        for(int i=0; i<nRows; i++){
            for(int j=0; j<len; j++){
                matrix[i][j]=0;
            }
        }
        int colum=0;    //当前放到了第colum列
        int row = 0;    //当前放到了第row行
        bool down = true;   //向下还是向上走
        for(int i=0; i<len; i++){
            matrix[row][colum]=s[i];
            if(down){
                row++;
                if(row==nRows){
                    row=row-2;
                    colum++;
                    down=false;
                }
            }else{
                row--;
                colum++;
                if(row<0){
                    row=1;
                    colum--;
                    down=true;
                }
            }
        }
        string result="";
        for(int i=0; i<nRows; i++){
            for(int j=0; j<len; j++){
                if(matrix[i][j]!=0){
                    result+=matrix[i][j];
                }
            }
        }
        return result;
    }
};

class Solution {
public:
    string convert(string s, int nRows) {
        if(nRows<2){
            return s;
        }
        int len=s.length();
        int step = 2*nRows-2;
        string result="";
        for(int i=0; i<nRows; i++){
            for(int j=i; j<len; j=j+step){
                result += s[j];
                if(i!=0 && i!=nRows-1 && j+step-2*i<len){   //除了第一行和最后一行，两列之间都有一个字符，其下标为j+step-2*i
                    result += s[j+step-2*i];
                }
            }
        }
        return result;
    }
};

[LeetCode] ZigZag Conversion

标签：c++   leetcode   

原文地址：http://blog.csdn.net/kangrydotnet/article/details/45096037