1129 Recommendation System (25 分)难度: 一般 / 知识点: 滑动窗口

Posted 2021-12-20 辉小歌

https://pintia.cn/problem-sets/994805342720868352/problems/994805348471259136
只需维护窗口即可。

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int cnt[N],n,k;
int s[11];
bool cmp(int a,int b)

    if(cnt[a]!=cnt[b]) return cnt[a]>cnt[b];
    return a<b;

int main(void)

    cin>>n>>k;
    int index=0;
    for(int i=0;i<n;i++)
    
        int id; cin>>id;
        if(i)
        
            printf("%d:",id);
            for(int j=0;j<index;j++)
            
                printf(" %d",s[j]);
            
            cout<<endl;
        
        bool flag=false;
        for(int j=0;j<index;j++)
        
            if(s[j]==id) flag=true;
            if(flag) break;
        
        cnt[id]++;
        if(!flag) s[index++]=id;//说明该次增加的商品不在窗口中，我们需要添加该窗口
        sort(s,s+index,cmp);//排序
        index=min(index,k);//维护串口
    
    return 0;