[Leetcode] 6. Zigzag Conversion

Zigzag Conversion - LeetCode

 

각 row 별로 순서대로 글자를 채우고 마지막에 하나로 합쳤다.

row 가 1인 경우도 제대로 작동할 줄 알았는데, 두 번째 줄로 들어가 버려 오답처리가 됐다.

class Solution {
public:
    string convert(string s, int numRows) 
    {
        vector<string> zig(numRows);
        int n = 1;
        int direct = 1;
        string ans;
        
        
        if(numRows == 1)
            return s;
        
        zig[0].push_back(s[0]);
        
        for(int i=1; i<s.size(); i++)
        {
            
            
            zig[n].push_back(s[i]);
            
            
            if( 0 < n && n < numRows - 1)
            {
                n += direct;
            }
            else
            {
                direct *= -1;
                n += direct;
            }
                
        }
        
        for(int k=0; k<numRows; k++)
        {
            ans += zig[k];
        }
        return ans;
    }
};