[LeetCode] ZigZag Conversion

zigzag模式字符串转换算法

最新推荐文章于 2021-11-05 22:58:08 发布

原创最新推荐文章于 2021-11-05 22:58:08 发布 · 503 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #C++

leetcode 同时被 2 个专栏收录

44 篇文章

订阅专栏

C++

19 篇文章

订阅专栏

本文介绍了一种将给定字符串转换为指定行数的zigzag模式的方法，并提供了两种实现方式，一种是直接操作字符串，另一种是使用二维数组。重点讨论了zigzag模式的形成规律及其在实际应用中的价值。

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)

	0	1	2	3	4	5	6
0	P		A		H		N
1	A	P	L	S	I	I	G
2	Y		I		R

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

And then read line by line: "PAHNAPLSIIGYIR"

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

string convert(string text, int nRows)

convert("PAYPALISHIRING", 3) should return "PAHNAPLSIIGYIR"

Idea: 通过找规律，可以知道一个zigzag 有 2 * nRows -2 个元素。

先放第一排和最后一排的元素。

在放之字形中间的元素

class Solution {
public:
    string convert(string s, int nRows) {
        int size = s.size();
        if(size <= 1 || nRows <= 1 || size <= nRows) return s;
        
        string result ;
        
        int k = 0,
            zigzag = 2 *nRows - 2, //one zigzag element count
            count = size/zigzag + 1;   //how many zigzag
        for(int i = 0; i < nRows; i++)
        {
            for(int j = i; j < size; j+=zigzag)
            {
                result.append(1, s[j]); // first and last
                if(i != 0 && i!= nRows -1 && j + zigzag - 2 *i < size)
                {
                    result.append(1, s[j + zigzag - 2 * i]);
                }
            }
        }
        
        return result;
    }
};

Idea: 将一维数组string[size] 转为二维数组result[nRows][ ], 然后按照行输出。

--------------------------------Memory Limit Exceeded------------------------------------------

class Solution {
public:
    string convert(string s, int nRows) {
        int size = s.size();
        if(size <= 1 || nRows <= 1 || size <= nRows) return s;
        
        int **p = new int* [nRows];
        int nCols = (size/(2*nRows - 2) + 1)*(nRows -1);
        for(int i = 0; i < nRows; i++)
        {
            p[i] = new int [nCols];
        }
        
        int k = 0,
            row = nRows - 1;
        for(int j = 0; j < nCols; j++)
        {
            for(int i = 0; i < nRows; i++)
            {
                if(!(j % row) && k < size)
                {
                    p[i][j] = s[k++];
                }
                else if(!(i + j % row) && k < size)
                {
                    p[i][j] = s[k++];
                }
                else
                {
                    p[i][j] = ',';
                }
            }
        }
        
        k = 0;
        for(int i = 0; i < nRows; i++)
        {
            for(int j = 0; j < nCols; j++)
            {
                if(p[i][j] != ',')
                {
                    s[k++] = p[i][j];
                    
                }
            }
        }
        
        for(int i = 0; i < nRows; i++)
        {
            delete [] p[i];
        }
        
        delete [] p;
        
        return s;
    }
};