T-shirt buying
A new pack of n t-shirts came to a shop. Each of the t-shirts is characterized by three integers pi, ai and bi, where pi is the price of the i-th t-shirt, ai is front color of the i-th t-shirt and bi is back color of the i-th t-shirt. All values piare distinct, and values ai and bi are integers from 1 to 3.
m buyers will come to the shop. Each of them wants to buy exactly one t-shirt. For the j-th buyer we know his favorite color cj.
A buyer agrees to buy a t-shirt, if at least one side (front or back) is painted in his favorite color. Among all t-shirts that have colors acceptable to this buyer he will choose the cheapest one. If there are no such t-shirts, the buyer won't buy anything. Assume that the buyers come one by one, and each buyer is served only after the previous one is served.
You are to compute the prices each buyer will pay for t-shirts.
Input
The first line contains single integer n (1 ≤ n ≤ 200 000) — the number of t-shirts.
The following line contains sequence of integers p1, p2, ..., pn (1 ≤ pi ≤ 1 000 000 000), where pi equals to the price of the i-th t-shirt.
The following line contains sequence of integers a1, a2, ..., an (1 ≤ ai ≤ 3), where aiequals to the front color of the i-th t-shirt.
The following line contains sequence of integers b1, b2, ..., bn (1 ≤ bi ≤ 3), where biequals to the back color of the i-th t-shirt.
The next line contains single integer m (1 ≤ m ≤ 200 000) — the number of buyers.
The following line contains sequence c1, c2, ..., cm (1 ≤ cj ≤ 3), where cj equals to the favorite color of the j-th buyer. The buyers will come to the shop in the order they are given in the input. Each buyer is served only after the previous one is served.
Output
Print to the first line m integers — the j-th integer should be equal to the price of the t-shirt which the j-th buyer will buy. If the j-th buyer won't buy anything, print -1.
Examples
Input
5
300 200 400 500 911
1 2 1 2 3
2 1 3 2 1
6
2 3 1 2 1 1
Output
200 400 300 500 911 -1
Input
2
1000000000 1
1 1
1 2
2
2 1
Output
1 1000000000
题意:给你每件衣服的价格,前后颜色(颜色都用数字表示),然后有 m 个顾客,每个顾客都有自己想要的颜色。如果有这样颜色的衣服(前后都行),输出 最小的价格。。。没有 就输出-1;
思路:可以用一维优先队列维护每种颜色的衣服的价格 以及 编号。。。,每一次提取队首的时候都判断 这件衣服是不是已经被买过了,被买过了就弹出队首。。
AC代码:
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxn = 2e5 + 10;
struct Pair{
int val,id;
Pair(){}
Pair(int val,int id):val(val),id(id){}
friend bool operator < (const Pair &A,const Pair & B){
return A.val > B.val;
}
};
priority_queue<Pair>Q[3];
int vis[maxn];
int p[maxn];
int a[maxn];
int b[maxn];
int main()
{
int n;
scanf("%d",&n);
for(int i = 1;i <= n;i ++) scanf("%d",&p[i]);
for(int i = 1;i <= n;i ++) scanf("%d",&a[i]);
for(int i = 1;i <= n;i ++) scanf("%d",&b[i]);
for(int i = 1;i <= n;i ++){
Q[a[i]].push(Pair(p[i],i));
Q[b[i]].push(Pair(p[i],i));
}
int m; scanf("%d",&m);
while(m --){
int c; scanf("%d",&c);
while(!Q[c].empty() && vis[Q[c].top().id])
Q[c].pop();
if(Q[c].empty()) printf("-1 ");
else {
printf("%d ",Q[c].top().val);
vis[Q[c].top().id] = 1;
Q[c].pop();
}
}
return 0;
}