zoukankan      html  css  js  c++  java
  • CodeForces413B

    B. T-shirt buying
    time limit per test
    3 seconds
    memory limit per test
    256 megabytes
    input
    standard input
    output
    standard output

    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 pi are 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 ai equals to the front color of the i-th t-shirt.

    The following line contains sequence of integers b1, b2, ..., bn (1 ≤ bi ≤ 3), where bi equals 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 

    有n件T恤,告诉你T恤价格p(并且价格互不相同),正面颜色a,背面颜色b(只有三种颜色)。

    有m位顾客,顾客要求的颜色c只要(c==a||c==b),在此情况下顾客会选择最便宜的一件,同时输出其价格,若没有满足条件的则输出-1.

    n和m的最大值为200000,n*m直接做必然会超时。

    最好的方法是将价格和颜色捆绑起来,按价格大小排序,不过对于set容器的运用不够熟练所以做不到。

    注意到颜色只有1,2,3这三种,所以我们可以用三个set代表三种颜色来储存价格,也算一种”另类捆绑“了。

    #include<bits/stdc++.h>
    using namespace std;
    
    int main()
    {
        int n,i,x,m;
        set<int> s1,s2,s3;
        set<int>::iterator it;
        int p[200005];
        cin>>n;
        for(i=0;i<n;i++)
            cin>>p[i];
        for(i=0;i<n;i++)
        {
            cin>>x;
            if(x==1)
                s1.insert(p[i]);
            else if(x==2)
                s2.insert(p[i]);
            else
                s3.insert(p[i]);
        }
        for(i=0;i<n;i++)
        {
            cin>>x;
            if(x==1)
                s1.insert(p[i]);
            else if(x==2)
                s2.insert(p[i]);
            else
                s3.insert(p[i]);
        }
        cin>>m;
        for(i=0;i<m;i++)
        {
            cin>>x;
            if(x==1)
            {
                if(s1.size()>0)
                {
                    it=s1.begin();
                    cout<<*it<<" ";
                    s2.erase(*it);
                    s3.erase(*it);
                    s1.erase(it);
                }
                else
                    cout<<"-1 ";
            }
            else if(x==2)
            {
               if(s2.size()>0)
                {
                    it=s2.begin();
                    cout<<*it<<" ";
                    s1.erase(*it);
                    s3.erase(*it);
                    s2.erase(it);
                }
                else
                    cout<<"-1 ";
            }
            else
            {
                if(s3.size()>0)
                {
                    it=s3.begin();
                    cout<<*it<<" ";
                    s1.erase(*it);
                    s2.erase(*it);
                    s3.erase(it);
                }
                else
                    cout<<"-1 ";
            }
        }
        return 0;
    }

    注意一下容器的清空顺序即可,下面附上第一种方法的代码

    #include<iostream>
    #include<vector>
    #include<set>
    #define MAX 200000
    using namespace std;
    int main()
    {
        int n,m,i,x,c[MAX+5];
        cin>>n;
        vector<int> p(n);
        vector< set<int> > v(4);
        for(i=0;i<n;i++)
            cin>>p[i];
        for(i=0;i<n;i++)
            cin>>x,v[x].insert(p[i]);
        for(i=0;i<n;i++)
            cin>>x,v[x].insert(p[i]);
        cin>>m;
        for(i=0;i<m;i++)
            cin>>c[i];
        for(i=0;i<m;i++)
        {
            if(!v[c[i]].empty())
            {
                n=*v[c[i]].begin(),cout<<n<<" ";
                for(x=1;x<4;x++)
                    v[x].erase(n);
            }
            else cout<<"-1 ";
        }
     return 0;
    }
  • 相关阅读:
    Oracle的导入导出
    Android利用数据库传送数据
    Android相对布局实例
    git拉取github项目
    kafka支持认证SASL_PLAINTEXT
    fluentd插件开发
    解决go依赖包安装问题
    ES/Kibana支持search-guard认证
    logkit docker运行和代码下载
    CentOS安装指定git版本
  • 原文地址:https://www.cnblogs.com/qq936584671/p/6847055.html
Copyright © 2011-2022 走看看