Hk148
Hk148

Reputation: 11

How to transfer arrayList elements into a 2D array?

I have an ArrayLists called List<Pair> patternList = new ArrayList<>() that contains a number of patterns of length 2. for example, for A={1,2,3,4} I create some patterns like (1,3)(3,2)...etc. I want to put these patterns into a 2d array(matrix)[A][A] in way that the patterns must go into a specific index in the matrix, For instance pattern (1,3) must go into index [3][3] or pattern (3,2) must go in [2][2].

Thanks.

Upvotes: 0

Views: 94

Answers (2)

Mshnik
Mshnik

Reputation: 7032

If you just want to put them in by the second index. (edited to allow multiple per index):

List<Pair>[][] matrix = new LinkedList<Pair>[5][5]; //Replace 5 here a getter for the maximum value of A
                                  //Here it's 5 because the max of your example is 4.
//Initialize all positions of matrix to empty list.
for (int r = 0; r < matrix.length; r++){
    for(int c = 0; c < matrix[r].length; c++){
        matrix[r][c] = new LinkedList<Pair>();
    }
}
for (Pair p : patternList){
    if (matrix[p.second][p.second] == null)
        matrix[p.second][p.second] = new LinkedList<Pair>();
    matrix[p.second][p.second].add(p);
}

This isn't really using the 2d aspect of the array though; you could accomplish exactly the same thing with a 1d array of lists:

List<Pair>[] arr = new LinkedList<Pair>[5]; //Replace 5 here a getter for the maximum value of A
                                  //Here it's 5 because the max of your example is 4.
//Initialize all positions of array to empty list.
for (int r = 0; r < matrix.length; r++){
    arr[r] = new LinkedList<Pair>();
}
for (Pair p : patternList){
    if (arr[p.second] == null)
        arr[p.second] = new LinkedList<Pair>();
    arr[p.second].add(p);
}

Upvotes: 1

John B
John B

Reputation: 32949

Since you want to keep more than one element at each position, I suggest you use a Multimap. A Multimap is basically a Map<Key, Collection<Value>> that takes care of creating the collections for you.

Multimap

 Multimap<List<Integer>, Pair> myMap = ArrayListMultimap.create();

 for (Pair p : myPairs){
     List<Integer> key = Lists.newArrayList(p.second, p.second);
     myPair.put(key, p);
 }

Upvotes: 0

Related Questions