特别说明:
感谢大佬IamOliverLew,这个算法是从他那里学来的。原始代码在这里
集装箱运输货物时,我们必须特别小心,不能把不相容的货物装在一只箱子里。比如氧化剂绝对不能跟易燃液体同箱,否则很容易造成爆炸。
本题给定一张不相容物品的清单,需要你检查每一张集装箱货品清单,判断它们是否能装在同一只箱子里。
输入格式:
输入第一行给出两个正整数:N (≤104) 是成对的不相容物品的对数;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
解题思路:
我自己原来的想法是先构造一个数据结构,用于保存每种货物的冲突货物,在装箱的时候,每来一个新货物就和已经在集装箱里的货物一一比对,确定没有冲突再装箱,结果时间复杂度太高,至少O(n^3)了,超时的问题无法解决。后来就想能不能原封不动的利用题目给出的成对的冲突物品的数据,但不知道怎么处理。结果看到了IamOliverLew的算法,让人眼前一亮,把每个集装箱的货物先排序,再去遍历之前的成对冲突货物清单,如果出现了一对冲突货物,就不能装箱,反之就可以。换了一个思路,问题瞬间就变得简单了!真是厉害啊!
#include <stdio.h>
#include <stdlib.h>
typedef struct _node {
int right;
int left;
} Pair;
int compare(const void *a, const void *b) {
return *(int*)a - *(int*)b;
}
int main(int argc, const char *argv[]) {
int N, M, K, i, flag;
if ( scanf("%d %d", &N, &M)==EOF ) printf("error\n");
Pair pairs[N];
for ( i=0; i<N; ++i )
if ( scanf("%d %d", &pairs[i].left, &pairs[i].right)==EOF ) printf("error\n");
while ( M-- ) {
if ( scanf("%d", &K)==EOF ) printf("error\n");
int cargos[K];
for ( i=0; i<K; ++i )
if ( scanf("%d", &cargos[i])==EOF ) printf("error\n");
qsort(cargos, K, sizeof(int), compare);
for ( i=0, flag=0; i<N; ++i ) {
if ( bsearch(&pairs[i].left, cargos, K, sizeof(int), compare) && bsearch(&pairs[i].right, cargos, K, sizeof(int), compare) ) {
flag = 1;
break;
}
}
printf("%s\n", flag ? "No" : "Yes");
}
return EXIT_SUCCESS;
}