《Cracking the Coding Interview》——第14章：Java——题目6

2014-04-26 19:11

题目：设计一个循环数组，使其支持高效率的循环移位。并能够使用foreach的方式访问。

解法：foreach不太清楚，循环移位我倒是实现了一个，用带有偏移量的数组实现。修改元素不一定能做到O(1)时间，但循环移位能在O(1)时间解决。不得不说，用不熟的语言写面试题，很难~~~

代码：

// 14.6 Implement a circular array, which allows easy rotation and array access.
// Combination of a vector and a rotation index.
import java.io.PrintStream;
import java.util.Vector;

public class CircularArray<T> {
    public int rotateIndex;
    public Vector<T> v;

    public CircularArray() {
        v = new Vector<T>();
        rotateIndex = 0;
    }

    public T elementAt(int index) {
        if (index < 0 || index > v.size() - 1) {
            throw new ArrayIndexOutOfBoundsException();
        }
        return v.elementAt((index + rotateIndex) % v.size());
    }

    public void add(T val) {
        if (v.size() > 0) {
            v.insertElementAt(val, (rotateIndex + v.size() - 1) % v.size() + 1);
            if (rotateIndex > 0) {
                ++rotateIndex;
            }
        } else {
            v.insertElementAt(val, 0);
        }
    }

    public void removeElementAt(int index) {
        if (rotateIndex > 0) {
            if (index < 0 || index > v.size() - 1) {
                throw new ArrayIndexOutOfBoundsException();
            }
            int tmp = v.size();
            v.removeElementAt((index + rotateIndex) % v.size());
            if (index >= tmp - rotateIndex && index <= tmp - 1) {
                --rotateIndex;
            }
        } else {
            v.removeElementAt(index);
        }
    }

    public void rotate(int index) {
        if (v.size() == 0) {
            return;
        }
        index = (v.size() - (v.size() - index) % v.size()) % v.size();
        rotateIndex = (rotateIndex + index) % v.size();
    }

    public static void main(String[] args) {
        CircularArray<Integer> c = new CircularArray<Integer>();
        PrintStream cout = System.out;

        c.add(3);
        c.add(4);
        cout.println(c.v);
        c.add(7);
        cout.println(c.v);
        c.rotate(2);
        c.add(12);
        c.add(25);
        cout.println(c.v);
        c.rotate(-1);
        c.add(77);
        cout.println(c.v);
        c.removeElementAt(2);
        cout.println(c.v);
        cout.println(c.elementAt(0));
    }
}