CCF认证201909-4推荐系统

题目描述

201909-4推荐系统

算法设计+题目分析

写在前面的
首先，跟看这篇blog的同学说下，本文代码为0分代码，但是，经过作者的测试，是能够过用例的，零分原因在于超时（题目要求5s，使用了5.03秒）

超时分析
1、sort函数的使用，本文在查询操作中共使用了1+1+m次，当各类的总数大于K时，对于每一类都用了排序，一次排序的复杂度约为O(nlogn)
2、vector.erase()的使用，复杂度约为O(n)

可能可以解决超时的办法
1、本文的排序是使用了vector+sort的办法，或者可以用set，应能更快的排序
2、关于erase的问题，思考了一下，由于类别+编号可以唯一确定一个商品，类别<=50，商品<=3*10^4，可以采用如下的办法舍得复杂度为O(1)

const long long mul = (long long)(1e9);
cin >> kind >> id;
long long a = kind * mul + id;
commodities.erase(um[a]);

为什么不解决
解决的话要重新建立结构，时间不划算。。。

算法设计+题目分析

分析
1、初始时，各类商品的编号和分数都一样！
2、用1、2、3分别代表三种操作
3、输出时，分数相同取类别编号小的，类别相同，取商品编号小的
4、没有选中的输出-1
5、最后一行注意不要换行

设计
1、建立一个node 结构，来存储kind,id,score
2、使用vector commodity[50]来给各类建立一个数组

几个核心点
sort的使用是核心

注意点+记录点

1. 注意每次输出时判断是否是最后一样，非最后一行的再换行（即最后一行不换行）
2. sort函数的使用

#include <algorithm>
sort(a, b, cmp)

a为数组起始地址，b为终止地址的下一个地址
3、vector的函数erase的使用

vector.erase(vector.begin()+k）

删除单个元素，即第0+k个

代码（0分）

环境
Visual C++6.0

#include <iostream>
#include <vector>
#include <algorithm>
#include <stdio.h>

using namespace std;
struct node{
	int kind, id, score;
};
vector<node> commodity[50];
bool cmp(node a, node b){//初级比较，返回分数大大
	if (a.score==b.score) return a.id<b.id;//分数一致时，返回编号小大
	return a.score>b.score;
}
bool cmp1(node a, node b){//按照编号从小到大排序
	return a.id<b.id;
}
void output(int op, int op_t, int m, vector<node> K_num, vector<int> num){//第一种简单输出，输出各类之和小于K大情况
		int l=0;
		for (int i=0; i<m; i++){//遍历各类
			if (num[i]==0){//如果没有推荐的商品
				int z=-1;
				if (op_t==op && i==m-1) cout<<z;//最后一行，不要换行
				else cout<<z<<endl;
				continue;
			}
			sort(K_num.begin()+l, K_num.begin()+l+num[i], cmp1);//按类排序
			int j=l;
			for (l; j<l+num[i]-1; j++){
				cout<<K_num[j].id<<" ";
			}
			if (op_t==op && i==m-1) cout<<K_num[l+num[i]-1].id;//最后一行，不要换行
			else cout<<K_num[l+num[i]-1].id<<endl;
			l+=num[i];
		}
}
bool cmp3(node a, node b){//在各类之和大于K的时候，实现题目中的排序要求
	if (a.score==b.score && a.kind==b.kind) return a.id<b.id;
	if (a.score==b.score && a.id==b.id) return a.kind<b.kind;
	return a.score>b.score;
	
}
void output2(int op, int op_t, int m, int total, vector<node> K_num){//各类之和大于K的时候的输出
	sort(K_num.begin(), K_num.end(), cmp3);//利用第三种分类对所有推荐的进行排序
	vector<node> temp[50];
	for(int i=0; i<total; i++){//按照类别放入相应的temp[i]
		temp[K_num[i].kind].push_back(K_num[i]);
	}
	for(int j=0; j<m; j++){//遍历各类
		if (temp[j].size()==0){
			int z=-1;
			if (op_t==op && j==m-1) cout<<z;//最后一行，不要换行
			else cout<<z<<endl;
			continue;
		}
		sort(temp[j].begin(), temp[j].end(), cmp1);类内排序
		for(int k=0; k<temp[j].size()-1; k++){
			cout<<temp[j][k].id<<" ";
		}
		if (op_t==op && j==m-1) cout<<"last"<<temp[j][temp[j].size()-1].id;//最后一行，不换行
		else cout<<temp[j][temp[j].size()-1].id<<endl;
	}


}
int main(){
	freopen("C:\\input.txt", "r", stdin);
	int m, n;
	cin>>m>>n;
	int id, score, op_num;
	for (int i=0; i<n; i++){//遍历每个商品
		cin>>id>>score;
		node product;
		product.id=id;
		product.score=score;
		for (int j=0; j<m; j++){//遍历各类
			product.kind=j;
			commodity[j].push_back(product);
		}
	}
	cin>>op_num;
	int c;
	int kind;
	for (int ii=0; ii<op_num; ii++){
		cin>>c;
		if (c==1){//增加商品
			cin>>kind>>id>>score;
			node product_c;
			product_c.kind=kind;
			product_c.id=id;
			product_c.score=score;
			commodity[kind].push_back(product_c);
		}
		if (c==2){//删除商品
			cin>>kind>>id;
			int k=0;
			for(; k<commodity[kind].size(); k++) {
				if(commodity[kind][k].id == id) break;
			}
			commodity[kind].erase(commodity[kind].begin()+k);
		}
		if (c==3){
			int total, num_t;
			cin>>total;
			vector<node> K_num;
			vector<int> num;
			for(int iii=0; iii<m; iii++){
				cin>>num_t;
				sort(commodity[iii].begin(), commodity[iii].end(), cmp);//类内排序
				for (int jj=0; jj<num_t && jj<commodity[iii].size(); jj++){
					K_num.push_back(commodity[iii][jj]);//选出各类前面k_i个商品
				}
				if (num_t>commodity[iii].size()) num.push_back(commodity[iii].size());//确定每类选出的商品数量
				else num.push_back(num_t);

			}
			if (total>=K_num.size()) output(op_num, ii+1, m, K_num, num);//在给类中挑选最大的num个
			if (total<K_num.size())	output2(op_num, ii+1, m, total, K_num);//一起挑选K个（total）

		}
	}
	return 0;
}

感谢链接

无