【算法】模拟算法——Z字形变换(medium)

题解：模拟算法——Z字形变换(medium)

1.题目

题目链接：LINK
在这里插入图片描述

2.题解

利用模拟，来解决问题。

首先创建出一个O(numRows*n)的数组来，并按照题目要求把每个字符按顺序填进去。

这里以numRows = 4，字符串s = "abcdefghijk"为例来演示如下：
在这里插入图片描述
然后我们按每行挨个把字符加进去就行了，除了很浪费空间…

所以，我们可以总结规律来进行优化：
在这里插入图片描述
规律可以分为两部分：
第一部分是第一行和最后一行，满足如下特点：
下标从numRows-1开始，且后一个比前一个多d
第二部分是中间那些行，满足如下特点：
两两一组,下标从{k,d-k}开始，下一组比前一组多d

图解如下：
在这里插入图片描述

3.参考代码

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

        int n = s.size();
        string ret;
        int d = 2*numRows - 2;
        
        //先处理第一行
        for(int i = 0; i < n; i+=d)
        {
            ret+=s[i];
        }
        //再处理中间一行
        for(int i = 1; i < numRows - 1; i++)//标识行
        {
            for(int j = i,k = d-i;j < n || k < n;j+=d,k+=d)//这个地方为什么用||来判定是否结束？防止一个条件满足了，另一个不满足从而导致漏字符的情况
            {
                if(j < n) ret+=s[j];//上面判断结束条件有可能是越界的，因而在加入之前应该先判断一下
                if(k < n) ret+=s[k];
            }
        }
        //处理最后一行
        for(int i = numRows-1; i < n; i+=d)
        {
            ret+=s[i];
        }

        return ret;
    }
};