LeetCode-6. Z字形變換（找規律）

kewlgrl發表於2018-05-02
LeetCode

6. Z字形變換

將字串 "PAYPALISHIRING" 以Z字形排列成給定的行數：

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

之後從左往右，逐行讀取字元："PAHNAPLSIIGYIR"

實現一個將字串進行指定行數變換的函式:

string convert(string s, int numRows);

示例 1:

輸入: s = "PAYPALISHIRING", numRows = 3
輸出: "PAHNAPLSIIGYIR"

示例 2:

輸入: s = "PAYPALISHIRING", numRows = 4
輸出: "PINALSIGYAHRPI"
解釋:

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




#include<bits/stdc++.h>
using namespace std;
/********************提交程式碼********************/
char* convert(char* s, int numRows)
{
    if(numRows==1) return s;//特殊處理1
    int n=strlen(s),i=0,cnt=0;
    int x=4+2*(numRows-3),y=0,z=0,t=0;
    bool flag=true;//x和y均非零時true取x,false取y
    char *ans=(char*)malloc((n+1)*sizeof(char));
    for(; i<numRows; ++i)
    {
        ans[cnt++]=s[i];
        t=i;
        while(true)
        {
            if(flag)
                z=x;
            else
                z=y;
            if(x==0)//特殊處理第一行和最後一行
                z=y;
            else if(y==0)
                z=x;
            if(t+z>=n) break;//跳出條件
            ans[cnt++]=s[t+z];
            flag=!flag;
            t+=z;
        }
        x-=2,y+=2,z=0;
        flag=true;
    }
    ans[cnt]='\0';
    return ans;
}
/***************************************************/
int main()
{
#ifdef ONLINE_JUDGE
#else
    freopen("F:/cb/read.txt","r",stdin);
    //freopen("F:/cb/out.txt","w",stdout);
#endif
    ios::sync_with_stdio(false);
    cin.tie(0);
    char s[1000],ss[10];
    while(cin.getline(s,1000))
    {
        int numRows;
        cin>>numRows;
        cout<<s<<endl;
        cout<<numRows<<endl;
        cout<<"/"<<convert(s,numRows)<<"/"<<endl;
        cin.getline(ss,10);//吃回車符
    }
    return 0;
}


當作找規律來做的。

下標化成數字後Z型排列如下：

numRows=3時：


第一行數字之間相差4；

第二行數字之間相差2；

第三行數字之間相差4。

numRows=4時：


第一行數字之間相差6；

第二行數字之間相差4，2；

第三行數字之間相差2，4。

第四行數字之間相差6。

numRows=5時：


第一行數字之間相差8；

第二行數字之間相差6，2；

第三行數字之間相差4，4。

第四行數字之間相差2，6；

第五行數字之間相差8。

……………………分隔線……………………

依給定的行數，按行遍歷，找規律發現第一行和最後一行差值是4+2*(numRows-3)

豎著看，每行之間兩個差值相差2

x和y分別表示某行兩個相鄰數的差值（為保持一致性，第一行和最後一行加個0）

交替取x和y，用flag標記當前取x還是y

相關文章