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

输出样例1

输入样例2

5
1   2   3   4   5
1   3   5   4   2
5   4   3   2   1
4

输出样例2

限制

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;
}