CCF 试题编号： 201909-4 试题名称： 推荐系统

这题是stl的综合应用，map要想快，直接上unordered_map，这样查询接近O(1)，是不是很嗨皮。

思路其实还是很简单的，type+id做个Hash，由于set.insert的第一个返回值是指向该插入元素的迭代器，所以，对于每一个type+id我们都可以存下它对应的迭代器，这样删除不就很快了吗，省去查找。

这题是我第一次用c++11的语法， 原谅我的low，嘻嘻，auto还挺好用。

#include <bits/stdc++.h>
using namespace std;

const long long BASE=1e10;

struct Pair {
    long long score,hash_val;
    Pair(long long s,long long h) {
        score=s;
        hash_val=h;
    }
    bool operator < (const Pair &b) const {
        if (score!=b.score) {
            return score>b.score;
        }
        long long id1=hash_val%BASE,id2=b.hash_val%BASE;
        long long type1=hash_val-id1,type2=b.hash_val-id2;
        if (type1!=type2) {
            return type1<type2;
        }
        return id1<id2;
    }
};

set<Pair> s;
// set<pair<long long,long long> > del;
//unordered_map<long long,long long>mp;
unordered_map<long long,set<Pair>::iterator >mp;
long long num[55],top[55],total,totaltop;
vector<long long> ans[55];
long long M,N;

long long Hash(long long type,long long id)
{
	return type*BASE+id;
}

void select()
{
    memset(num,0,sizeof(num));
    long long cnt=0;
//    for (set<Pair>:: iterator it=s.begin();it!=s.end()&&cnt<total;it++) {
//        long long id=it->hash_val%BASE,type=it->hash_val/BASE;
//
////        printf("SELECT: \n");
////        printf("type: %I64d id: %I64d\n",type,id);
//        if (num[type]<top[type]) {
//            cnt++;
//            num[type]++;
//            ans[type].push_back(id);
////            printf("IN:\n");
////            printf("type: %I64d id: %I64d\n",type,id);
//        }
//        else continue;
//    }

    for (auto it:s) {
        long long id=it.hash_val%BASE,type=it.hash_val/BASE;
        if (num[type]<top[type]) {
            num[type]++;
            cnt++;
            ans[type].push_back(id);
        }
        if (cnt==total||cnt==totaltop) break;
    }
//    printf("ANS:\n");
//    for (int i=0;i<M;i++) {
//        for (int j=0;j<ans[i].size();j++) {
//            printf("%I64d ",ans[i][j]);
//        }
//        puts("");
//    }
//    puts("");
}

void print() {
    for (set<Pair>::iterator it=s.begin();it!=s.end();it++) {
        printf("sc:%I64d type:%I64d id:%I64d\n",it->score,it->hash_val/BASE,it->hash_val%BASE);
    }
    puts("");
}

int main()
{
//    long long hash_val=Hash(12,23);
//    printf("%I64d %I64d\n",hash_val%BASE,hash_val-hash_val%BASE);
//
//    s.insert(Pair(5,Hash(2,1)));
//    s.insert(Pair(5,Hash(2,2)));
//    s.insert(Pair(6,Hash(7,2)));
//    mp[Hash(1,1)]=s.insert(Pair(5,Hash(1,1))).first;
//    s.erase(mp[Hash(1,1)]);
//    for (set<Pair>::iterator it=s.begin();it!=s.end();it++) {
//        printf("%lld %lld %lld\n",it->score,it->hash_val%BASE,it->hash_val-it->hash_val%BASE);
//    }
    freopen("in.txt","r",stdin);
    long long sc,id,opnum,op,type,comm;
    scanf("%I64d%I64d",&M,&N);
    for (int i=0;i<N;i++) {
        scanf("%I64d%I64d",&id,&sc);
        for (int j=0;j<M;j++) {
            mp[Hash(j,id)]=s.insert(Pair(sc,Hash(j,id))).first;
        }
    }
//    print();
    scanf("%I64d",&opnum);
    for (int i=0;i<opnum;i++) {
        scanf("%I64d",&op);
        if (op==1) {
            scanf("%I64d%I64d%I64d",&type,&comm,&sc);
            mp[Hash(type,comm)]=s.insert(Pair(sc,Hash(type,comm))).first;
//            print();
        }
        else if (op==2) {
            scanf("%I64d%I64d",&type,&comm);
            s.erase(mp[Hash(type,comm)]);
//            print();
        }
        else {
            scanf("%I64d",&total);
            totaltop=0;
            for (int j=0;j<M;j++) {
                scanf("%I64d",&top[j]);
                totaltop+=top[j];
                ans[j].clear();
            }
//            print();
            select();
            for (int j=0;j<M;j++) {
                if (ans[j].size()==0) {
                    printf("-1\n");
                }
                else {
                    int anssize=ans[j].size();
//                    for (int k=0;k<anssize;k++){
//                        printf("%I64d",ans[j][k]);
//                        if (k==anssize-1) putchar('\n');
//                        else putchar(' ');
//                    }
                    for (int i=0;i<anssize;i++) {
                        printf("%I64d",ans[j][i]);
                        if (i==anssize-1) {
                            putchar('\n');

                        }
                        else {
                            putchar(' ');
                        }
                    }
                }
                ans[j].clear();
            }
        }
    }
    return 0;
}