【Leetcode 6】 ZigZag Conversion

原创于 2017-05-22 16:07:10 发布 · 325 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #C #算法 #algorithm

LeetCode 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种将字符串按照N字型排列的算法实现，通过动态数组记录每行的字符，最终输出重新排列后的字符串。该算法适用于特定格式的字符串处理任务。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给定一个字符串，按照N字型的写法吧字符串重新排列，然后按从上到下，从左到右输出新的字符串。（下面是例子）

"PAYPALISHIRING"

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

答案："PAHNAPLSIIGYIR"

思路：开一个动态数组，对于一个字符只要知道他现在要放在第几行就可以了。不需要知道具体坐标，因为最后空格位置是不输出的。只需要开 nrow个 vector 记录对应行加入的字符，再从上往下输出即可。

代码如下：

class Solution {
public:
    string convert(string s, int numRows) {
    	vector<char>e[numRows];
    	int now=0,flag=1;
    	if(numRows==1)return s;
    	for(int i=0;i<s.length();++i){
    		e[now].push_back(s[i]);
    		if(now==0){
    			now++;
    			flag=1;
    			continue;
    		}
    		if(now==numRows-1){
    			now--;
    			flag=0;
    			continue;
    		}
    		if(flag==1)
    			now++;
    		else
    			now--;
    	}
    	string ans="";
    	for(int i=0;i<numRows;++i){
    		for(int j=0;j<e[i].size();++j)
    			ans+=e[i][j];
    	}
    	return ans;
    }
};