leetcode 6. ZigZag Conversion

发表于 | 更新于

题目链接：

https://leetcode.com/problems/zigzag-conversion/#/description

题意：

给一个字符串，要求将字符串排列成锯齿状，然后按行从左到右输出。如下图，原来的字符串顺序为： BFGAHIDJKCLME，按行读就是BDEFIJMGHKLAC。
图示

题解：

找规律即可，按行来看相邻两个点的距离分为两个，假设为a和b，第 i 行为[2(n-i-1)，2i]，第一行相当于b为0，第二行相当于a为0，距离为零表示两点重合，不考虑。依次输出。

代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#include <bits/stdc++.h>
using namespace std;

class Solution
{
public:
    string convert(string s, int numRows)
    {
        if(numRows==1) return s;
        string ans="";
        for(int i=0; i<numRows; i++)
        {
            int a=2*(numRows-i-1),b=2*i;
            for(int j=i; j<s.size();)
            {
                if(a){
                    ans+=s[j];
                    j+=a;
                }
                if(j>=s.size()) break;
                if(b){
                    ans+=s[j];
                    j+=b;
                }
            }
        }
        return ans;
    }
};
int main()
{
    Solution sol;
    string str;
    int n;
    while(cin>>str>>n){
        cout<<sol.convert(str,n)<<endl;
    }
    return 0;
}