PTA 乙级 1090 危险品装箱

#include<stdio.h>
#include<stdlib.h>

typedef struct
{
	int t1, t2;
}struction1;

typedef struct
{
	int *array, cnt;
}struction2;

int main()
{
	int M, N, count[100000];

	scanf("%d %d", &N, &M);
	struction1 *ban = (struction1*)malloc(sizeof(struction1)*N);
	for (int i = 0; i < 100000; i++)
		count[i] = 0;
	for (int i = 0; i < N; i++)
		scanf("%d %d", &ban[i].t1, &ban[i].t2),
		count[ban[i].t1]++, count[ban[i].t2]++;
	struction2 *a = (struction2*)malloc(sizeof(struction2) * 100000);
	for (int i = 0; i < 100000; i++)
		a[i].array = (int*)malloc(sizeof(int)*count[i]), a[i].cnt = count[i];
	for (int i = 0; i < N; i++)
		a[ban[i].t1].array[--count[ban[i].t1]] = ban[i].t2,
		a[ban[i].t2].array[--count[ban[i].t2]] = ban[i].t1;
	for (int i = 0; i < M; i++)
	{
		int K, *tools, flag = 1;
		scanf("%d", &K);
		tools = (int*)malloc(sizeof(int)*K);
		for (int j = 0; j < K; j++)
			scanf("%d", &tools[j]);
		for (int j = 0; j < K; j++)
		{
			for (int k = 0; k < a[tools[j]].cnt; k++)
			{
				for(int p=j+1;p<K;p++)
					if (a[tools[j]].array[k] == tools[p])
					{
						flag = 0; break;
					}
				if (flag == 0)
					break;
			}
			if (flag == 0)
				break;
		}
		if (flag == 0)
			printf("No\n");
		else printf("Yes\n");
	}

	return 0;
}