[LeetCode/릿코드] 118. Pascal's Triangle

문제

 

 

반응형

---

풀이

 

 

 

import java.util.ArrayList;
import java.util.List;
class Solution {
    public List<List<Integer>> generate(int numRows) {
        List<List<Integer>> answer = new ArrayList<List<Integer>>();
 
        List<Integer> first = new ArrayList<Integer>();
        first.add(1);
        
        List<Integer> second = new ArrayList<Integer>();
        second.add(1);
        second.add(1);
 
        if(numRows == 1) {
            answer.add(first);
            return answer;
        };
        if(numRows == 2) {
            answer.add(first);
            answer.add(second);
            return answer;
        }
 
        answer.add(first);
        answer.add(second);
        for(int i=3; i<=numRows; i++) {
            List<Integer> temp = new ArrayList<Integer>();
            for(int j=1; j<=i; j++) {
                if(j == 1 || j == i) {
                    temp.add(1);
                } else {
                    int n = answer.get(i-1-1).get(j-1-1) + answer.get(i-1-1).get(j-1);
                    temp.add(n);
                }
            }
            answer.add(temp);
        }
        return answer;
    }
}