PAT (Advanced Level) Practice - 1026 Table Tennis（30 分）

博客围绕PAT (Advanced Level) Practice的1026题Table Tennis展开。介绍解题思路，如VIP客户选桌规则，客户和桌子分总队列与VIP队列。还提到实现时改用set，以及使用中总桌子和VIP桌子队的删除操作，同时指出存在丢失精度问题，并给出AC代码和Debug代码。

题目链接：点击打开链接

题目大意：略。

解题思路：

如果VIP客户遇到序号小的普通桌子和序号大的VIP桌子，优先选择VIP桌；如果VIP桌没有了，VIP客户和普通客户是平等的，VIP客户就不能插队了。

客户、桌子：总的一队，VIP的一队。

注意1：这里因为中途发现桌子用优先队列来做不方便，所以改用set来做：
VIP桌被占用后记得总桌子需要删除对应的VIP桌子；反之，总桌子在使用过程中，遇到使用VIP桌子时，在VIP桌子队中也要删除对应的VIP桌子。

注意2：(p.stime-p.atime)/60.0+1e-4，丢失精度问题。

AC 代码

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a)
#define ssclr(ss) ss.clear(), ss.str("")
#define INF 0x3f3f3f3f
#define MOD 1000000007

using namespace std;

typedef long long ll;

const int maxn=1e4+10;

struct peo
{
    int ah,am,as,sh,sm,ss;
    int atime,stime,ptime;
    int vip;
}ps[maxn];

struct player
{
    int tid,pid,etime;
    player(){}
    player(int tid,int pid,int etime):tid(tid),pid(pid),etime(etime){}
    friend bool operator<(player p1,player p2)
    {
        return p1.etime>p2.etime;
    }
};

priority_queue<player> pq;
set<int> tbpq, vtbpq;
vector<int> tbv, vipv;
int vis[maxn], cnt[200];

int cmp1(peo p1,peo p2)
{
    return p1.atime<p2.atime;
}

int cmp2(peo p1,peo p2)
{
    return p1.stime<p2.stime;
}

int main()
{
    int n,h,m,s,ptime,vip,k,kvip;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d:%d:%d%d%d",&h,&m,&s,&ptime,&vip);
        if(ptime>120) ptime=120;
        ps[i].ah=h, ps[i].am=m, ps[i].as=s, ps[i].ptime=ptime*60, ps[i].vip=vip;
        ps[i].atime=ps[i].ah*3600+ps[i].am*60+ps[i].as;
    }
    sort(ps,ps+n,cmp1);
    for(int i=0;i<n;i++) if(ps[i].vip==1) vipv.push_back(i);

    scanf("%d%d",&k,&kvip);
    tbv.resize(k+1);
    for(int i=1;i<=kvip;i++) scanf("%d",&vip), tbv[vip]=1;
    for(int i=1;i<=k;i++)
    {
        tbpq.insert(i);
        if(tbv[i]==1) vtbpq.insert(i);
    }

    int l1=0, l2=0;
    for(int t=28800; t<75600 && l1<n;)
    {
        while(l1<n && vis[l1]==1) l1++;
        if(l1==n) break;

        if(ps[l1].atime>t)
        {
            t=ps[l1].atime;
            continue;
        }
        else if(!pq.empty() && pq.top().etime>t && tbpq.empty())
        {
            t=pq.top().etime;
            continue;
        }

        while(!pq.empty()) // 清理 ps[l1].atime<=t
        {
            int f=0;
            player pla=pq.top();
            if(pla.etime<=t)
            {
                tbpq.insert(pla.tid);
                if(tbv[pla.tid]==1) vtbpq.insert(pla.tid);
                pq.pop();
                f=1;
            }
            if(!f) break;
        }
        if(tbpq.empty()) continue;

        while(1) // 有空位 + 有人
        {
            while(l1<n && vis[l1]==1) l1++;
            while(l2<vipv.size() && vis[vipv[l2]]==1) l2++;

            if(l1==n || tbpq.empty() || ps[l1].atime>t) break;

            int tid;
            if(!vtbpq.empty() && l2<vipv.size() && ps[vipv[l2]].atime<=t) // VIP席遇上对的人
            {
                auto it=vtbpq.begin();
                tid=*it, vtbpq.erase(it), tbpq.erase(tbpq.find(*it));
                ps[vipv[l2]].stime=t;
                pq.push(player(tid,vipv[l2],ps[vipv[l2]].ptime+t));
                vis[vipv[l2]]=1;
            }
            else
            {
                auto it=tbpq.begin();
                tid=*it, tbpq.erase(it);
                if(tbv[tid]==1)
                    vtbpq.erase(vtbpq.find(tid));
                ps[l1].stime=t;
                pq.push(player(tid,l1,ps[l1].ptime+t));
                vis[l1]=1;
            }
            cnt[tid]++;
        }
    }

    sort(ps,ps+n,cmp2);
    for(int i=0;i<n;i++)
    {
        peo &p = ps[i];
        if(p.stime==0) continue;
        printf("%02d:%02d:%02d %02d:%02d:%02d %.0f\n",p.ah,p.am,p.as,p.stime/3600,p.stime%3600/60,p.stime%60,(p.stime-p.atime)/60.0+1e-4);
    }
    for(int i=1;i<=k;i++) printf("%d%c",cnt[i],i==k?'\n':' ');

    return 0;
}

Debug 代码

#include<bits/stdc++.h>
#include<cmath>

#define mem(a,b) memset(a,b,sizeof a)
#define ssclr(ss) ss.clear(), ss.str("")
#define INF 0x3f3f3f3f
#define MOD 1000000007

using namespace std;

typedef long long ll;

const int maxn=1e4+10;

struct peo
{
    int ah,am,as,sh,sm,ss;
    int atime,stime,ptime;
    int vip;
}ps[maxn];

struct player
{
    int tid,pid,etime;
    player(){}
    player(int tid,int pid,int etime):tid(tid),pid(pid),etime(etime){}
    friend bool operator<(player p1,player p2)
    {
        return p1.etime>p2.etime;
    }
};

priority_queue<player> pq;
set<int> tbpq, vtbpq;
vector<int> tbv, vipv;
int vis[maxn], cnt[200];

int cmp1(peo p1,peo p2)
{
    return p1.atime<p2.atime;
}

int cmp2(peo p1,peo p2)
{
    return p1.stime<p2.stime;
}

void show(int len)
{
    puts("-------------");
    for(int i=0;i<len;i++)
    {
        peo &p=ps[i];
        printf("%02d:%02d:%02d %d %d %d\n",p.ah,p.am,p.as,p.ptime/60,p.vip,p.atime);
    }
    puts("-------------");
}

void showPQ()
{
    puts("-------------");
    if(!pq.empty())
    {
        int et=pq.top().etime;
        printf("%02d:%02d:%02d\n",et/3600,et%3600/60,et%60);
    }
    puts("-------------");
}

int main()
{
    int n,h,m,s,ptime,vip,k,kvip;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%d:%d:%d%d%d",&h,&m,&s,&ptime,&vip);
        if(ptime>120) ptime=120;
        ps[i].ah=h, ps[i].am=m, ps[i].as=s, ps[i].ptime=ptime*60, ps[i].vip=vip;
        ps[i].atime=ps[i].ah*3600+ps[i].am*60+ps[i].as;
    }
    sort(ps,ps+n,cmp1);
    for(int i=0;i<n;i++) if(ps[i].vip==1) vipv.push_back(i);

    scanf("%d%d",&k,&kvip);
    tbv.resize(k+1);
    for(int i=1;i<=kvip;i++)
    {
        scanf("%d",&vip);
        tbv[vip]=1;
    }
    for(int i=1;i<=k;i++)
    {
        tbpq.insert(i);
        if(tbv[i]==1) vtbpq.insert(i);

    }

//    show(n);

    int l1=0, l2=0;
    for(int t=28800; t<75600 && l1<n;)
    {
//        printf("-------\n%02d:%02d:%02d\n-------\n",t/3600,t%3600/60,t%60);
        // 原理上一个一个接下来是不会有失效的，这么做是避免碰到VIP已经使用了的客户
        while(l1<n && vis[l1]==1)
            l1++;
        if(l1==n) break;

        if(ps[l1].atime>t)
        {
            t=ps[l1].atime;
            continue;
        }
        else if(!pq.empty() && pq.top().etime>t && tbpq.empty())
        {
            t=pq.top().etime;
            continue;
        }

        // ps[l1].atime<=t
        while(!pq.empty()) // 清理
        {
            int f=0;
            player pla=pq.top();
            if(pla.etime<=t)
            {
                tbpq.insert(pla.tid);
                if(tbv[pla.tid]==1) vtbpq.insert(pla.tid);
                pq.pop();
                f=1;
            }
            if(!f) break;
        }
        if(tbpq.empty()) continue;

        while(1) // 有空位 + 有人
        {
            while(l1<n && vis[l1]==1)
                l1++;
            while(l2<vipv.size() && vis[vipv[l2]]==1)
                l2++;

            if(l1==n) break;

            if(tbpq.empty() || ps[l1].atime>t) break;

            int tid;
            if(!vtbpq.empty() && l2<vipv.size() && ps[vipv[l2]].atime<=t) // VIP席遇上对的人
            {
                auto it=vtbpq.begin();
                tid=*it, vtbpq.erase(it), tbpq.erase(tbpq.find(*it));
                ps[vipv[l2]].stime=t;
                pq.push(player(tid,vipv[l2],ps[vipv[l2]].ptime+t));
                vis[vipv[l2]]=1;
            }
            else
            {
                auto it=tbpq.begin();
                tid=*it, tbpq.erase(it);
                if(tbv[tid]==1)
                    vtbpq.erase(vtbpq.find(tid));
                ps[l1].stime=t;
                pq.push(player(tid,l1,ps[l1].ptime+t));
                vis[l1]=1;
            }
            cnt[tid]++;

//            showPQ();
        }
    }

    sort(ps,ps+n,cmp2);
    for(int i=0;i<n;i++)
    {
        peo &p = ps[i];
        if(p.stime==0) continue;
        printf("%02d:%02d:%02d %02d:%02d:%02d %.0f\n",p.ah,p.am,p.as,p.stime/3600,p.stime%3600/60,p.stime%60,(p.stime-p.atime)/60.0+1e-4);
    }
    for(int i=1;i<=k;i++) printf("%d%c",cnt[i],i==k?'\n':' ');

    return 0;
}

/* 特殊样例
3
19:29:59 75 1
08:08:42 33 1
08:24:14 120 1
3 1
2 
*/