小红书24秋招后端开发编程题（牛客）

最近没事的时候在牛客上找笔试题刷着玩，发现网上题解不多，发出来记录一下，欢迎大家一起交流学习

2. 小红书推荐系统

题目链接：小红书校招笔试真题_C++工程师、golang工程师_牛客网

小红书有一个推荐系统，可以根据用户搜索的关键词推荐用户希望获取的内容。
现在给定小红的搜索记录（记录为分词后的结果），我们认为当一个单词出现的次数不少于3次时，该单词为“用户期望搜索的单词”，即称为关键词。请你根据小红的记录，输出小红的用户画像对应的所有关键词。

思路：统计每个字符串出现的次数，按照长度排序，若长度相同则按照字符串的升序排序

#include<bits/stdc++.h>
using namespace std;
map<string, int> mp;
vector<string> v;
bool cmp (string s1, string s2) {
    if (mp[s1] == mp[s2]) {
        return s1 < s2;
    }
    else return mp[s1] > mp[s2];
}
int main()
{
    string s;
    while(cin>>s) {
        mp[s] += 1;
        if (mp[s] == 1) {
            v.push_back(s);
        }
    }
    sort(v.begin(), v.end(), cmp);
    for (size_t i=0; i<v.size(); i++) {
        if (mp[v[i]] >= 3) printf("%s\n", v[i].c_str());
        // if (mp[v[i]] >= 3) cout<<v[i]<<endl;
        else break;
    }
    return 0;
}

15. 小红的分享日常

题目链接：小红书校招笔试真题_C++工程师、golang工程师_牛客网

小红很喜欢前往小红书分享她的日常生活。已知她生活中有i个事件，分享第i个事件需要她花费ti的时间和​的精力来编辑文章，并能获得ai的快乐值。
小红想知道，在总花费时间不超过T且总花费精力不超过H的前提下，小红最多可以获得多少快乐值？

思路：经典的背包问题，比传统的01背包加了一个维度，直接加就行，在每次循环中更新答案，注意如果当前是第一条数据，需要初始化答案

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll f[60][510][510];
ll x[60], y[60], z[60];
int main()
{
    ll n;
    cin>>n;
    ll t, h;
    cin >> t >> h;
    for (ll i=0; i<n; i++) cin >> x[i] >> y[i] >> z[i];
    ll ans = 0;
    for (ll i=0; i<=n; i++)
    {
        for (ll j=0; j<=t; j++)
        {
            for(ll k=0; k<=h; k++)
            {
                if (i==0 && !(j < x[i] || k < y[i])) f[i][j][k] = z[i];
                else if (i == 0) continue;
                else if (j < x[i] || k < y[i]) f[i][j][k] = f[i-1][j][k];
                else f[i][j][k] = max(f[i-1][j][k], f[i-1][j-x[i]][k-y[i]] + z[i]);
                // cout << i << j << k << endl;
                // cout << f[i][j][k] <<endl;
                ans = max(ans, f[i][j][k]);
            }
        }
    }
    cout << ans <<endl;
    return 0;
}

26. 小红的小红树

题目链接：小红书校招笔试真题_C++工程师、golang工程师_牛客网

小红在刷小红书的时候看到了一颗挂着小红薯的小红树，所以小红也想种一颗小红树挂一些小红薯发小红书。
小红有一颗树，每个结点有一个权值，初始时每个节点都是白色。小红每次操作可以选择两个相邻的结点，如果它们都是白色且权值的和是质数，小红就可以选择其中一个节点染红。
小红想知道最多可以染红多少个节点？

思路：贪心地考虑在每次符合条件的染色中染儿子节点，因为儿子节点可以染多个，父亲节点染色后与其关联的儿子节点都不能被染色，实现的时候需要在dfs回溯的时间执行染色操作。

#include<bits/stdc++.h>
using namespace std;
vector<int> G[100010];
int a[100010];
int vis[100010];
int col[100010];
int res;
bool is_p(int x) {
    int sq = sqrt(x);
    if (x == 2) return true;
    for (int i=2; i<=sq; i++) {
        if (x % i == 0) return false;
    }
    return true;
}
void dfs(int rt) {
    vis[rt] = 1;
    for (int i=0; i<G[rt].size(); i++) {
        int dis = G[rt][i];
        if (vis[dis] == 1) continue;
        dfs(dis);
        if (is_p(a[dis] + a[rt])) {
            if (col[rt] == 0) {
                col[dis] = 1;
                // cout << rt << " " << dis<< endl;
                // cout << a[dis] + a[rt] << endl;
                res++;
            }
        }
    }
}
int main()
{
    int n;
    scanf("%d", &n);
    for(int i=1; i<=n; i++) scanf("%d", &a[i]);
    for(int i=0; i<n; i++) {
        int u,v;
        scanf("%d %d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1);
    printf("%d\n", res);
    return 0;
}