题海战(set)

题目描述
某信息学奥赛教练经验丰富,他的内部题库有 m 道题。他有 n 个学生,第 i 个学生已经做过p[i]道题。由于马上要进行noip考试,该教练准备举行 k 场比赛和训练。每场比赛或训练都会有一些他的学生参加,但是如何选题令他非常烦恼。对于每场比赛,他要保证所出的题没有任何一道已有任何一个学生做过;而对于每场训练,他要保证所出的所有题都被每一个参赛学生做过。

输入格式
第1行2个正整数n和m,表示学生数和题库中的题目总量。

第2~n+1行,先是1个正整数p,然后p个整数表示第i个学生的做题记录(可以重复做同一道题)。

第n+2行,1个正整数k,表示要举行比赛和训练的总场数(可能有学生重复报名)。

接下来的k行,每行的第1个整数type表示是训练或者比赛(1为训练,0为比赛)。第二个数q表示参赛学生数,然后q个正整数表示参赛学生编号。每一行的两个数之间有一个空格。

输出格式
共k行,每行表示本次训练或比赛可选的最多题目(由小到大排序,中间用一个空格隔开,如果没有输出一个空行)。

样例
样例输入
5 10
2 3 7
1 3
2 4 7
3 3 6 10
7 1 2 3 4 7 8 9
6
0 3 3 4 5
0 3 1 3 4
1 2 1 3
0 1 5
1 1 2
1 2 3 5
样例输出
5
1 2 5 8 9
7
5 6 10
3
4 7
数据范围与提示
1 ≤ n,k ≤ 100
1 ≤ m ≤ 10000
题解:
看到:“可能有学生重复报名”,“可以重复做同一道题”,就知道要去重了;
众所周知:
c++去重那家强;
STL找set.
这道题没什么难度;
详细见代码:

#include<bits/stdc++.h>
using namespace std;
int a,b,n[10005],m[105][105],k[10005],c,d,e,m1[105][105],n1[10005];
set<int> st;
int main(){
    
    
	scanf("%d%d",&a,&b);
	for(int i=1;i<=a;i++){
    
    
		st.clear();
		scanf("%d",&n[i]);
		for(int j=1;j<=n[i];j++){
    
    
			scanf("%d",&m[i][j]);
			st.insert(m[i][j]);
		}
		n[i]=0;
		int w[10005]={
    
    };
		for(set<int>::iterator it=st.begin();it!=st.end();it++){
    
    
			n[i]++;
			m[i][n[i]]=*it;//set去重
			w[*it]=1;
		}
		for(int l=1;l<=b;l++){
    
    
			if(w[l]==0){
    
    
				n1[i]++;
				m1[i][n1[i]]=l;//存一下没有做的题有哪些
			}
		}
	}
	scanf("%d",&c);
	for(int i=1;i<=c;i++){
    
    
		st.clear();
		scanf("%d %d",&d,&e);
		for(int j=1;j<=e;j++){
    
    
			scanf("%d",&k[j]);
			st.insert(k[j]);
		}
		int j=0;
		for(set<int>::iterator it=st.begin();it!=st.end();it++){
    
    
			j++;
			k[j]=*it;//set去重
		}
		if(d==1){
    
    
			int w[10005]={
    
    };
			for(int l=1;l<=j;l++){
    
    
				for(int p=1;p<=n1[k[l]];p++){
    
    
					w[m1[k[l]][p]]=1;//标记不能做的题
				}
			}
			for(int l=1;l<=b;l++){
    
    
				if(w[l]==0){
    
    
					printf("%d ",l);//输出能做的题
				}
			}
			printf("\n");
		}
		else{
    
    
			int w[10005]={
    
    };
			for(int l=1;l<=j;l++){
    
    
				for(int p=1;p<=n[k[l]];p++){
    
    
					w[m[k[l]][p]]=1;//标记不能做的题
				}
			}
			for(int l=1;l<=b;l++){
    
    
				if(w[l]==0){
    
    
					printf("%d ",l);//输出能做的题
				}
			}
			printf("\n");
		}
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/cqbz_xsh/article/details/107549427
set