THU2015 fall 1-1 Team
描述
教练员A、B和C将要从编号为1到n的队员中挑选自己的队员。为公平起见,每个教练都根据自己的喜好程度将队员排序;你负责根据以下规则为他们分配队员。
你拿到的数据是a、b、c三个数组,表示三个教练对队员的喜好程度排序,每个数组都是数字1到n的一个排列,下标越小表示教练越喜欢该队员。你的分组规则是,从还未被分配的队员中找一个教练A最喜欢的队员分到A组;然后,在未分配的队员中分配教练B最喜欢的队员到B组;然后是教练C;再是教练A、B......依次类推直到所有队员分配完毕。
现在队员k希望知道自己被分配给哪位教练,请你来告诉他。
输入
共5行。
第1行包含一个整数n。
第2至4行依次是数组a、b和c,每行都是整数[1, n]的一个排列。
第5行包含一个整数k。
输出
仅一个字符,A、B或C,表示队员k被分配给哪位教练。
输入样例1
3
1 2 3
1 2 3
1 2 3
3
输出样例1
C
输入样例2
5
1 2 3 4 5
1 3 5 4 2
5 4 3 2 1
4
输出样例2
B
限制
1 <= n <= 500,000
1 <= k <= n
时间:1 sec
空间:256 MB
提示
一级提示
大体上,1 sec内,O(n)的算法可以通过n = 10,000,000规模的数据,O(nlogn)通过500,000规模,O(n^2)通过5,000规模。
本题等一些复杂度是O(n)的题目受限于scanf(“%d”, ...)的读入速度,但又不希望通过读取二进制文件等不直观的方式增加同学们的负担。
代码如下:
#include <stdio.h>
#include <string.h>
const int SZ = 1 << 20; //提升IO buff
struct fastio{
char inbuf[SZ];
char outbuf[SZ];
fastio(){
setvbuf(stdin, inbuf, _IOFBF, SZ);
setvbuf(stdout, outbuf, _IOFBF, SZ);
}
}io;
#define N 500001
int dp[4][N];
int mark[N]; // 0:未选 1:已选
char ret[N];
void team(int n, int *mark)
{
int a = 1, b = 1, c = 1;
int t;
for (int i = 0; i < n; i++)
{
t = i % 3;
switch (t)
{
case 0:
while (mark[dp[t][a]] == 1) a++;
mark[dp[t][a]] = 1;
ret[dp[t][a]] = 'A';
break;
case 1:
while (mark[dp[t][b]] == 1) b++;
mark[dp[t][b]] = 1;
ret[dp[t][b]] = 'B';
break;
case 2:
while (mark[dp[t][c]] == 1) c++;
mark[dp[t][c]] = 1;
ret[dp[t][c]] = 'C';
break;
}
}
}
int main()
{
int n;
int k;
scanf("%d", &n);
for (int i = 0; i < 3; i++)
{
for (int j = 1; j <= n; j++)
scanf("%d", &dp[i][j]);
}
memset(mark, 0, n);
team(n, mark);
scanf("%d", &k);
printf("%c\n", ret[k]);
return 0;
}