昨天看的一道笔试题,靠回忆写下题意。
题意:合并多个数组,第一次从第一个数组中读取n个值,第二次从第二个数组读取n个,依次往下,然后回到第一个数组继续如此读,读过的值不能在读,数组不足n个,读剩下的,直到所有数组数读完
第一行输入n
第2-n行输入一组数,用逗号隔开
输出一个数组,打印,用逗号隔开
输入示例:
3
1,2,3,9,6,8,7
8,7,8,6,1,6,9
输出示例:
1,2,3,8,7,8,9,6,8,6,1,6,7,9
自己只是帮着看的题,自己在eclipse上,输入几组用例,都能正常通过,不知道算法复杂度或全部用例能不能达标,仅做参考。
参考代码:
public class Main {
public static void main(String[] args) throws InterruptedException, IOException {
BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
ArrayList<String> list = new ArrayList<String>();
String s = "";
while(!(s=bf.readLine()).equals("")) {//每次读一行,不输入时,回车换行结束
list.add(s);
}
int n = Integer.valueOf(list.get(0));//读取第一个数
int [] array = Sort(n,list);
for(int i = 0;i<array.length;i++) {//输出最终结果
if(i == array.length-1) System.out.print(array[i]);
else System.out.print(array[i]+",");
}
}
public static int [] Sort(int n,ArrayList<String> s) {
final ArrayList<ArrayList> list = new ArrayList<ArrayList>();
for(int i = 1;i<s.size();i++) {//将每组数用逗号隔开,添加进ArrayList集合中,
// 多组ArrayList在添加进外面一层list集合
ArrayList temp = new ArrayList();
String str = s.get(i);
String [] strs = str.split(",");
for(int j =0;j<strs.length;j++) {
temp.add(strs[j]);
}
list.add(temp);
}
ArrayList<Integer> result = new ArrayList();//保存最终的结果
while(true) {
Iterator it = list.iterator();//遍历最外面一层list集合
while(it.hasNext()) {
ArrayList array = (ArrayList) it.next();
Iterator it2 = array.iterator();//遍历里面的小list集合
if(array.size()<n) {//判断是否还有n个数
while(it2.hasNext()) {
result.add(Integer.valueOf((String) it2.next()));
}
it.remove();//不足n个数后,添加完直接删除该集合
}else {
int j =0;
while(j++<n){
result.add(Integer.valueOf((String) it2.next()));
it2.remove();//删除遍历过的数
}
}
}
if(list.size() ==0) break;
}
int [] array3 = new int[result.size()];//最终返回一个数组
for(int i=0;i<result.size();i++) {
array3[i] = result.get(i);
}
return array3;
}
}
感觉自己思路没问题,肯定有大佬写算法复杂度更低的,继续学习中。