Codeforces Round #413 (Div.1 + Div. 2) B. T-shirt buying

最新推荐文章于 2021-10-19 20:48:49 发布

拔光祖国小草

最新推荐文章于 2021-10-19 20:48:49 发布

阅读量323

点赞数

分类专栏： codeforces 文章标签： codeforces

本文链接：https://blog.csdn.net/yxm980918/article/details/71725415

版权

codeforces 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

是一个模拟水题，但是写的时候WA了无数次最后改了还是TLE，而且很麻烦。。需要锻炼啊。

先给上我TLE的版本。留着以后思考为什么TLE

我原来的大体思路是：开三个数组分别代表1 、2 、3的三种衣服，排序。顾客来了从左往右扫。然后用t1 t2 t3分别代表之前顾客已经扫到的位置。是否卖给顾客这条衣服时，用一个used数组记录之前是否卖过，没有则可以，卖了继续往下扫直到没有。下一个顾客来了从上一次更新的t1 t2 t3扫。

-1的情况是衣服卖完了或者没有这种衣服。用sum记录是否衣服卖完。

#include <cstdio>
#include <iostream>
#include <string>
#include <algorithm>
#include <map>
#include <cstring>
using namespace std;
const int maxn = 200005;
map <long long ,int> maps;
bool used[maxn];
long long cost[maxn];
long long  a[maxn];
long long  b[maxn];
long long  c[maxn];
int t1 = 0, t2 = 0, t3 = 0;
int cnt1 = 0, cnt2 = 0, cnt3 = 0;
int sum = 0;
void solve(long long *x, int t,int ccc)
{
    if(t <= ccc && ccc)
    {
        if(!used[maps[x[t]]])
        {
            used[maps[x[t]]] = true;
            cout << x[t];
            t++;
            sum++;
        }
        else
        {
            bool flag = false;
            while(used[maps[x[t]]])
            {
                t++;
                if(t>=ccc)
                {
                    cout << "-1";
                    flag = true;
                    break;
                }
            }
            if(!flag)
            {
                used[maps[x[t]]] = true;
                cout << x[t];
                t++;
                sum++;
            }
        }
    }
    else
        cout << "-1";

}
int main()
{
    int n;
    while(cin >> n && n)
    {
        memset(used,false,sizeof(used));
        maps.clear();
        for(int i=0; i<n; i++)
        {
            long long p;
            cin >> p;
            cost[i] = p;
            maps[p] = i;
        }
        cnt1 = 0;
        cnt2 = 0;
        cnt3 = 0;
        for(int i=0; i<n; i++)
        {
            int num;
            cin >> num;
            if(num == 1)
                a[cnt1++] = cost[i];
            else if(num == 2)
                b[cnt2++] = cost[i];
            else
                c[cnt3++] = cost[i];

        }
        for(int i=0; i<n; i++)
        {
            int num;
            cin >> num;
            if(num == 1)
                a[cnt1++] = cost[i];
            else if(num == 2)
                b[cnt2++] = cost[i];
            else
                c[cnt3++] = cost[i];
        }
        sort(a,a+cnt1);
        sort(b,b+cnt2);
        sort(c,c+cnt3);
        int tot;
        sum = 0;
        cin >> tot;
        t1 = 0, t2 = 0, t3 = 0;
        for(int i=0; i<tot; i++)
        {
            int num;
            cin >> num;
            if(i) cout << " ";
            if(sum > n) cout << "-1" ;
            else
            {
                if(num == 1)
                    solve(a,t1,cnt1);
                else if(num == 2)
                    solve(b,t2,cnt2);
                else
                    solve(c,t3,cnt3);
            }
        }
        cout << endl;
    }
    return 0;
}

然后是正确的做法。

用3个set集合存储三种衣服。因为set是可以自动排序的。然后从头开始扫，卖完了或者当前可以卖。扫完后便删除。

-1的情况是全部卖完。

（其实我觉得思路差不多的，只是用set更加的方便）

#include <cstdio>
#include <iostream>
#include <string>
#include <algorithm>
#include <map>
#include <cstring>
#include <set>
using namespace std;
const int maxn = 200000 + 5;
int p[maxn];
int a[maxn];
int b[maxn];
set<int> s[4];
map<int,int> maps;
bool used[maxn];
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF && n)
    {
        memset(used,false,sizeof(used));
        for(int i=0; i<n; i++)
        {
            scanf("%d", &p[i]);
            maps[p[i]] = i;
        }
        for(int i=0; i<n; i++)
        {
            scanf("%d",&a[i]);
            s[a[i]].insert(p[i]);
        }
        for(int i=0; i<n; i++)
        {
            scanf("%d",&b[i]);
            s[b[i]].insert(p[i]);
        }
        int m;
        scanf("%d",&m);
        for(int i=0; i<m; i++)
        {
            int ans = -1;
            int t;
            scanf("%d",&t);
            if(i) printf(" ");
            while(!s[t].empty())
            {
                if(!used[maps[*(s[t].begin())]])
                {
                    used[maps[*(s[t].begin())]] = true;
                    ans = *(s[t].begin());
                    s[t].erase(s[t].begin());
                    break;
                }
                else
                {
                    s[t].erase(s[t].begin());
                }
            }
            printf("%d",ans);

        }
        printf("\n");
    }

}