笔试题1——多数组合并

题目：现在有多组整数数组，需要将他们合并成一个新的数组。合并规则，从每个数组里按顺序取出固定长度的内容合并到新的数组中，取完的内容会删除掉，如果该行不足固定长度或者已经为空，则直接取出剩余部分的内容放到新的数组中，继续下一行。如样例1，获得长度3，先遍历第一行，获得2,5,6；再遍历第二行，获得1,7,4；再循环回到第一行，获得7,9,5；再遍历第二行，获得3,4；再回到第一行，获得7，按顺序拼接成最终结果。
输入描述：
第一行是每次读取的固定长度，长度>0；
第2-n行是需要合并的数组，不同的数组用回车换行分隔，数组内部用逗号分隔。
输出描述：
输出一个新的数组，用逗号分隔。
输入：
3
2,5,6,7,9,5,7
1,7,4,3,4
输出：
2,5,6,1,7,4,7,9,5,3,4,7
具体代码如下：

#include <iostream>
#include <string>
#include <stdio.h>
#include <queue>

using namespace std;

int main()
{
	queue<int> a[10000];
	int res[100005];
	int num;
	char ch;
	int cnt = 0;
	string str;

	int k;
	cin >> k;
	while(scanf("%d%c", &num, &ch) != EOF){
		a[cnt].push(num);
		if(ch == '\n'){
			cnt++;
		}
	}
	num = 0;
	while(true){
		int flag = 0;
		for (int i = 0; i < cnt; i++){ //cnt代表输入的数组的个数
			int l = 0;
			while(!a[i].empty() && l < k){
				flag = 1;
				int ans = a[i].front();
				a[i].pop();
				l++;
				res[num++] = ans;
			}
		}
		if(flag == 0)
			break;
	}

	for(int i = 0; i < num; i++){ //以 1，2，3，4 的格式输出合并后的新数组
		cout << res[i];
		if(i == num - 1)
			printf("\n");
		else
			printf(",");
	}

	return 0;
}