?LeetCode刷題實戰(zhàn)281：鋸齒迭代器

輸入:
v1 = [1,2]
v2 = [3,4,5,6] 
輸出: [1,3,2,4,5,6]

解析: 通過連續(xù)調用 next 函數(shù)直到 hasNext 函數(shù)返回 false，
     next 函數(shù)返回值的次序應依次為: [1,3,2,4,5,6]。
     
拓展：假如給你 k 個一維向量呢？你的代碼在這種情況下的擴展性又會如何呢?

public class ZigzagIterator {
    List<List<Integer>> data = new ArrayList<>(); //所有數(shù)據(jù)
    int size; //數(shù)據(jù)列表的總數(shù)
    int cur; //當前遍歷到的list的序號
    List<Integer> index = new ArrayList(); //每個list中當前遍歷到的位置index

    public ZigzagIterator(List<Integer> v1, List<Integer> v2) {
        data.add(v1);
        data.add(v2);
        size = 2; //目前只有兩個列表，可擴展為k個
        cur = 0; //從第0個list開始遍歷
        index.add(0);
        index.add(0);
    }

    public int next() {
        int ans = data.get(cur).get(index.get(cur)); //取出當前的數(shù)據(jù)
        index.set(cur,index.get(cur)+1); //更改位置到下一個數(shù)據(jù)，有效性有hasNext來保證
        cur = (cur+1) % size;
        return ans;
    }

    public boolean hasNext() {
        int start = cur; //保存目前的序號，用來判斷是否所有l(wèi)ist都已經遍歷到了最后一個位置
        while(index.get(cur)==data.get(cur).size()){
            cur = (cur+1) % size;

            /**
            運行到下一個if語句，說明所有的list都在最后一個位置，cur回到了最開始進入while循環(huán)的位置
            此時，所有數(shù)據(jù)都已經遍歷完成，返回false
            如果沒有這一句，會進入無限循環(huán)
            */
            if(cur == start) return false; 
        }
        return true;
    }
}