2016-10-24 3 views
4

Я изо всех сил пытаюсь создать правильную логику, чтобы сгладить массив. Я по сути хочу дублировать родительские строки для каждого дочернего элемента во вложенном массиве. Количество вложенных массивов может меняться. Я создаю списки Java bc. Я считаю, что их легко работать, но они открыты для любого решения. Характер этой проблемы заключается в том, что я начинаю с некоторого вложенного JSON, который я хочу преобразовать в плоский csv для загрузки в таблицу базы данных. Спасибо за помощь.Сгладить вложенные массивы. (Java)

Пример:

[1,2,[A,B,[Cat,Dog]],3] 

Я создал выше в списке. Каждый элемент представляет собой строку или другой список.

Результат:

[1,2,A,Cat,3], 
[1,2,A,Dog,3], 
[1,2,B,Cat,3], 
[1,2,B,Dog,3] 

Вот что я до сих пор. Очевидно, что не работает.

private static List<List<String>> processData(List<String> row, List<Object> data, List<List<String>> rowList) { 
    List<List<String>> tempRowList = new ArrayList<List<String>>(); 
    for (Object i : data) { 
     if (i instanceof List<?>) { 
      flattenArray((List<Object>) i, row, rowList); 
     } else { 
      for (List<String> r : rowList) { 
       r.add(i.toString()); //add item to all rows 
      } 
     } 
    } 
    return rowList; 

private static void flattenArray(List<Object> arrayObject, List<String> rowToCopy, List<List<String>> rowList) { 

    for (Object x : arrayObject) { 

     if (x instanceof List<?>) { 
      for (List<String> t : rowList) { 
       flattenArray((List<Object>) x, t, rowList); 
      } 

     } else { 
      List<String> newRow = new ArrayList<String>(rowToCopy); //copy row 

      List<Object> t = new ArrayList<Object>(); 
      t.add(x); 

      List<List<String>> innerRowList = new ArrayList<List<String>>(); 
      innerRowList.add(newRow); 

      processData(newRow, t, innerRowList); //pass in new copied row. object to add, 
      rowList.add(newRow); 

     } 

    } 
    rowList.remove(rowToCopy); 
} 

И я все устроил так.

public static void main(String[] args) { 

    List<Object> data = new ArrayList<Object>(); 
    List<List<String>> rowList = new ArrayList<List<String>>(); 

    data.add("1"); 
    data.add("2"); 

    List<Object> l1 = new ArrayList<Object>(); 
    l1.add("A"); 
    l1.add("B");   
    List<Object> l2 = new ArrayList<Object>(); 
    l2.add("dog"); 
    l2.add("cat"); 
    l1.add(l2); 
    data.add(l1); 

    data.add("3"); 

    List<String> r0 = new ArrayList<String>(); 
    rowList.add(r0); 
    System.out.println(data); 
    rowList = processData(r0, data, rowList); 
    System.out.println(rowList); 
} 
+1

Какой код вы писали? Давай увидим это? – Ash

+1

ОК, добавлен мой код. Я чувствую, что мне это удалось. – Andrew

+0

Это хорошо, теперь мы можем помочь решить, какие у вас проблемы – Ash

ответ

2

Я думаю, что это должно работать для вас, если вы используете Java 8:

List<Object> a = new ArrayList<>(); 
    List<Object> a1 = new ArrayList<>(); 
    a1.add("v"); 
    a1.add("w"); 
    List<Object> a2 = new ArrayList<>(); 
    a2.add("ww"); 
    a.add("a"); 
    a.add("b"); 
    a.add("c"); 
    a.add("d"); 
    a.add(a1); 
    a.add(a2); 
    List<Object> b = new ArrayList<>(); 

    a.stream().flatMap(x -> x instanceof String ? Stream.of(x) : ((List) x).stream()).forEach(b::add); 

    System.out.println(a); 
    System.out.println(b); 

b список будет содержать сплющенные a список. Выход:

[a, b, c, d, [v, w], [ww]] 
[a, b, c, d, v, w, ww] 
+0

Я использую Java 8. Я немного поиграл с потоками. Было любопытно, могут ли они помочь мне здесь. – Andrew

+0

Я не уверен, как их использовать, хотя на основе вашего примера. – Andrew

+0

[Andrew] (http://stackoverflow.com/users/6210427/andrew) это должно работать. Я знаю, что это некрасиво, я постараюсь превзойти его как можно скорее. – GROX13

Смежные вопросы