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

在这里插入图片描述
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;
}

Guess you like

Origin blog.csdn.net/qq_46527915/article/details/121525755