代码改变世界

Leetcode 6. ZigZag Conversion

2019-06-06 19:12  yx1989  阅读(170)  评论(0编辑  收藏  举报

这道题比较简单,观察清楚规律后很快可以写出来。

但是用go写的版本,总是答案错误,本地无问题。后用Java重写。

func convert(s string, numRows int) string {
	if numRows == 1 {
		return s
	}
	numChars := make(map[int][]string)
	size := 2 * numRows - 2
	for i := 0; i<len(s); i++ {
		x := s[i : i+1]
		if i % size < numRows {
			row := i % size
			numChars[row] = append(numChars[row], x)
		} else {
			row := size - i % size
			numChars[row] = append(numChars[row], x)
		}
	}

	var result string

	for _, ss := range numChars {
		for _, c := range ss {
			result += c
		}
	}

	return result
}
class Solution {
    public String convert(String s, int numRows) {
        if (numRows == 1 || s == null) {
            return s;
        }
        Map<Integer, List<String>> numChars = new HashMap<Integer, List<String>>();
        int size = 2 * numRows - 2;
        for (int i = 0; i < s.length(); i++) {
            String x = s.substring(i, i + 1);
            int row;
            if (i % size < numRows) {
                row = i % size;
            } else {
                row = size - i % size;
            }
            if (numChars.get(row) == null) {
                numChars.put(row, new ArrayList<String>());
            }
            numChars.get(row).add(x);
        }

        StringBuilder result = new StringBuilder();
        for (List<String> ss : numChars.values()) {
            for (String c : ss) {
                result.append(c);
            }
        }

        return result.toString();
    }
}