PAT (Basic Level) Practice 1090 危险品装箱 (25 分)

集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。

本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。

输入格式:

输入第一行给出两个正整数:N (≤10​4​​) 是成对的不相容物品的对数;M (≤100) 是集装箱货品清单的单数。

随后数据分两大块给出。第一块有 N 行,每行给出一对不相容的物品。第二块有 M 行,每行给出一箱货物的清单,格式如下:

K G[1] G[2] ... G[K]

其中 K (≤1000) 是物品件数,G[i] 是物品的编号。简单起见,每件物品用一个 5 位数的编号代表。两个数字之间用空格分隔。

输出格式:

对每箱货物清单,判断是否可以安全运输。如果没有不相容物品,则在一行中输出 Yes,否则输出 No

输入样例:

6 3
20001 20002
20003 20004
20005 20006
20003 20001
20005 20004
20004 20006
4 00001 20004 00002 20003
5 98823 20002 20003 20006 10010
3 12345 67890 23333

输出样例:

No
Yes
Yes

 算法一:(第三个测试点没过运行超时)

 代码实现:

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;

typedef struct Danger
{
    char a[6];
    char b[6];
} Danger;

int main()
{
    int n,m;
    int num,flag1,flag2;

    scanf("%d%d",&n,&m);
    Danger dan[n];
    for(int i = 0; i < n; i++)
    {
        scanf("%s%s",dan[i].a,dan[i].b);
        /*if(strcmp(dan[i].a,dan[i].b) > 0)
        {
            char temp[6];
            strcpy(temp,dan[i].a);
            strcpy(dan[i].a,dan[i].b);
            strcpy(dan[i].b,temp);
        }*/
    }
    for(int k = 0; k < m; k++)
    {
        scanf("%d",&num);
        char goods[num][6];
        for(int i = 0; i < num; i++)
            scanf("%s",goods[i]);
        /*for(int i = 0; i < num-1; i++)
        {
            int t = i;
            for(int j = i+1; j < num; j++)
            {
                if(strcmp(goods[j],goods[t]) < 0)
                {
                    t = j;
                }
            }
            if(t != i)
            {
                char temp[6];
                strcpy(temp,goods[i]);
                strcpy(goods[i],goods[t]);
                strcpy(goods[t],temp);
            }
        }*/

        /*for(int i = 0;i < num;i++)
            printf("%s ",goods[i]);
        printf("\n");
        for(int i = 0;i < n;i++)
            printf("%s %s\n",dan[i].a,dan[i].b);*/

        for(int i = 0; i < n;i++)
        {
            flag1 = 0,flag2 = 0;
            int j;
            for(j = 0;j < num;j++){
                if(strcmp(dan[i].a,goods[j]) == 0){
                    flag1 = 1;
                }
            }
            if(flag1 == 1&&flag2==0){
                for(j = 0;j < num;j++){
                    if(strcmp(dan[i].b,goods[j]) == 0){
                        flag2 = 1;
                    }
                }
            }
            if(flag1==1&&flag2==1)
                break;
        }
        //printf("flag1 = %d flag2 = %d\n",flag1,flag2);
        if(flag1==1&&flag2==1)
            printf("No\n");
        else
            printf("Yes\n");
    }
    return 0;
}

此时,for循环嵌套,时间复杂度为O(mn(num)) 

算法二:

继续优化,使用set集合存储货物清单,使用其成员函数find(),若找到,返回该键值迭代器的位置,否则,返回最后一个元素后面一个位置,时间复杂度为 O(logN) 。算法的时间复杂度为:O(mnlogK)测试点全过。

代码实现:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <set>
using namespace std;

int main()
{
    int n,m;
    int danger[100010][2];
    set<int> goods;
    scanf("%d%d",&n,&m);
    for(int i = 0;i < n;i++)
        scanf("%d%d",&danger[i][0],&danger[i][1]);
    for(int i = 0;i < m;i++){
        int k;
        int flag = 1;
        scanf("%d",&k);
        for(int j = 0;j < k;j++){
            int temp;
            scanf("%d",&temp);
            goods.insert(temp);
        }
        for(int j = 0;j < n;j++){
            if(goods.find(danger[j][0]) != goods.end() && goods.find(danger[j][1]) != goods.end()){//如果存在不相容的物品,立即跳出循环
                flag = 0;
                break;
            }
        }
        if(flag == 1)
            printf("Yes\n");
        else
            printf("No\n");
        goods.clear();//清空,进行下面货物的存储
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Annfan123/article/details/86520712
今日推荐