1. 程式人生 > >leetcode刷題6<z字形變換>

leetcode刷題6<z字形變換>

image 復雜 style clas round ring 變換 函數 mage

將一個給定字符串根據給定的行數,以從上往下、從左到右進行 Z 字形排列。

比如輸入字符串為 "LEETCODEISHIRING" 行數為 3 時,排列如下:

L   C   I   R
E T O E S I I G
E   D   H   N
之後,你的輸出需要從左往右逐行讀取,產生出一個新的字符串,比如:"LCIRETOESIIGEDHN"。

請你實現這個將字符串進行指定行數變換的函數:

string convert(string s, int numRows);
示例 1:

輸入: s = "LEETCODEISHIRING", numRows = 3
輸出: "LCIRETOESIIGEDHN
" 示例 2: 輸入: s = "LEETCODEISHIRING", numRows = 4 輸出: "LDREOEIIECIHNTSG" 解釋: L D R

開始沒有思路,看完官方的思路和代碼然後這樣做的

char* convert(char* s, int numRows) {
    if (numRows == 1||numRows==0)
        return s;
    int n=strlen(s);
    if(n==0||n==1||n==2) return s;
    char *re = (char *)malloc((n + 1) * sizeof
(char)); int j, k = 0; for (int i = 0; i < numRows; i++) { j = i; while (j < n) { re[k++] = s[j]; if (i != 0 && i != numRows - 1 && j + 2 * (numRows - 1 - i) < n) re[k++] = s[j + 2 * (numRows - 1 - i)]; j
+= 2 * numRows - 2; } } re[n] = \0; return re; }

但是這樣算法的復雜度有點高,大概是0(numRows+n)吧,後來看了下評論,發現一個比較優化的解法

,也是逐行處理,但是算法的復雜度貌似縮小到了0(n),大概思路是先遍歷0行,然後是i行,然後是剩下的

代碼如下:

char* convert(char* s, int numRows) {
    //獲取s長度
    if(numRows==1)return s;
    int len=strlen(s); 
    if(len==1||len==2) return s;
    //定義循環變量
    char* ret=(char *)malloc(sizeof(char)*(len+1));
    int i,j=0,k=0;
    //判斷不形成z字
    //定義指針變量
    for(i=0;i<len;i=k*(2*numRows-2))
    {
        ret[j++]=s[i];
        k++;
    }
    //判斷滿足那個條件
    for(i=1;i<len-1;i++)
    {
        k=0;
        while(k*(2*numRows-2)+i<len)
        {
            //按照索引添加
            ret[j++]=s[k*(numRows-2)+i];
            if((k+1)*(2*numRows-2)-i<len) ret[j++]=s[(k+1)*(2*numRows-2)-i];
         k++;
        }
        
    }
    k=0;
    for(i=numRows-1;i<len;i=k*(2*numRows-2)+numRows-1)
    {
        ret[j++]=s[i];
        k++;
    }
    ret[j]=\0;
    return ret;
}

leetcode刷題6<z字形變換>