solution.tar

src/org/blueocean/ClientRGAQ.java000644 000765 000024 00000002004 12412365602 020277 0ustar00junminliustaff000000 000000 package org.blueocean;

import java.util.ArrayList;
import java.util.List;

public class ClientRGAQ {
	/*
	 * generic class represent each element of the square
	 */
	static class Node<T>{
		private T data;
		
		public Node(T obj){
			this.data = obj;
		}
		
		public T getData(){
			return data;
		}
	}
	
	/*
	 * 2D array is a square
	 */
	public static <T> Node<T>[][] to2DArray(T[] input){
		int rowNum = (int) Math.ceil(Math.sqrt(input.length));
		@SuppressWarnings("unchecked")
		Node<T>[][] arrays = new Node[rowNum][rowNum];
		for(int i=0; i<input.length; i++){
			Node<T> node = new Node<T>(input[i]);
			arrays[i/rowNum][i%rowNum] = node;
		}	
		return arrays;
	}
	
	/**
	 * convert 2D array to list
	 * 
	 * @param input
	 * @return
	 */
	public static <T> List<T> to1DArray(Node<T>[][] input){
		List<T> resultList = new ArrayList<T>();
		for(int i=0; i<input.length; i++){
			for(int j=0; j<input[i].length; j++)
				if(input[i][j]!=null)
					resultList.add(input[i][j].data);
		}					
		return resultList;
	}
	

}
test/org/blueocean/ClientRGAQTest.java000644 000765 000024 00000003136 12412365544 021343 0ustar00junminliustaff000000 000000 package org.blueocean;

import java.util.List;

import org.blueocean.ClientRGAQ.Node;

import junit.framework.Assert;
import junit.framework.TestCase;

public class ClientRGAQTest  extends TestCase {
	
	public void testTo2DArray(){
		Node<String>[][] to2dArray = ClientRGAQ.to2DArray(new String[]{"A"});
		
		ClientRGAQ.Node<String>[][] result = to2dArray;
		
		Assert.assertEquals(result.length, 1);
		
		result = ClientRGAQ.to2DArray(new String[]{"A", "B"});
		
		Assert.assertEquals(result.length, 2);
		
		result = ClientRGAQ.to2DArray(new String[]{"A", "B", "C"});
		
		Assert.assertEquals(result.length, 2);
		
		result = ClientRGAQ.to2DArray(new String[]{"A", "B", "C", "D"});
		
		Assert.assertEquals(result.length, 2);
		
		result = ClientRGAQ.to2DArray(new String[]{"A", "B", "C", "D", "E"});
		
		Assert.assertEquals(result.length, 3);
		
		Assert.assertNotNull(result[0][1]);
		
		Assert.assertEquals(result[0][1].getData(), "B");
		
		Assert.assertNull(result[1][2]);
		
	}
	
	public void testTo1DArray(){
		Node<String>[][] to2dArray = ClientRGAQ.to2DArray(new String[]{"A"});
		
		List<String> t = ClientRGAQ.to1DArray(to2dArray);
		
		Assert.assertEquals(t.size(), 1);
		
		Assert.assertEquals(t.get(0), "A");
		
		to2dArray = ClientRGAQ.to2DArray(new String[]{"A", "B", "C", "D", "E"});
		
		t = ClientRGAQ.to1DArray(to2dArray);
		
		Assert.assertEquals(t.size(), 5);
		
		to2dArray = ClientRGAQ.to2DArray(new String[]{"A", "B", "C", null,"D", "E"});
		
		t = ClientRGAQ.to1DArray(to2dArray);
		
		Assert.assertNull(t.get(3));
		
		Assert.assertEquals(t.get(2), "C");
		
		Assert.assertEquals(t.size(), 6);
	}

}