【笔试】2022/08/11-zoom

  1. 一颗二叉树,节点是1-n,根节点是1,其他节点的顺序未知
    每个节点可能是红色或蓝色
    每个节点的权重=根到该节点的路径上两色之差,取绝对值
    求所有节点的权重和
    输入路径,但不知父子关系,如:
    12
    24
    31

我当时的答案,但是只过了30%

#include <iostream>
#include <vector>
#include <string>
#include <queue>

using namespace std;
using vi = vector<int>;

int main() {
  int n;
  int res = 0;
  cin >> n;
  string colors;
  cin >> colors;
  colors = '0' + colors;
  vector<vi> tree(n + 1);
  vi weight(n + 1);
  /**
   * 0 not visited
   * 1 red is more
   * 2 equal
   * 3 blue is more
  */
  vi visited(n + 1);
  for (int i = 1; i < n; i++) {
    int a, b;
    cin >> a >> b;
    tree[a].emplace_back(b);
    tree[b].emplace_back(a);
  }
  queue<int> q;
  q.push(1);
  if (colors[1] == 'R') {
    weight[1] = 1;
    visited[1] = 1;
  } else {
    weight[1] = -1;
    visited[1] = 3;
  }
  // weight[1] = colors[1] == 'R' ? 1 : -1;
  // visited[1] = 1;
  res = weight[1];
  while (!q.empty()) {
    int curr_node = q.front();
    // res += weight[curr_node];
    q.pop();
    for (int i = 0; i < tree[curr_node].size(); i++) {
      int son = tree[curr_node][i];
      if (visited[son]) continue;
      visited[son] = 1;
      if (colors[son] == 'R') {
        if (visited[curr_node] < 3) {
          weight[son] = weight[curr_node] + 1;
          visited[son] = 1;
        } else {
          weight[son] = weight[curr_node] - 1;
          if (weight[son] == 0) {
            visited[son] = 2;
          } else {
            visited[son] = 3;
          }
        }
      } else {
        if (visited[curr_node] == 1) {
          weight[son] = weight[curr_node] - 1;
          if (weight[son] == 0) {
            visited[son] = 2;
          } else {
            visited[son] = 1;
          }
        } else {
          weight[son] = weight[curr_node] + 1;
          visited[son] = 3;
        }
      }
      // cout << son << " " << weight[son] <<  endl;
      res += weight[son];
      q.push(son);
      // for (int j = 0; j < tree[son].size(); j++) {
      //   int grandson = tree[son][j];
      //   if (visited[grandson] == 0) {
      //     q.push(grandson);
      //   }
      // }
    }
  }
  cout << res;
  return 0;
}
  1. 输入如下格式
    1 Alice 2
    Zoom Apple
    2 Bob
    2 Alice
    1 Bob 2
    Apple Microsoft
    2 Bob
    1开头表示这是一个注册请求,占2行
    2表示这是一个查询请求,占1行
    要实现一个推荐系统,所有相关联的项目会组成一个集合,与之相关的人要收到推荐
    如开头表示:Alice关注2个,zoom和apple
    查询bob,不存在
    查询Alice,推荐个数0
    注册bob,关注2个,apple Microsoft
    这一动作组成了推荐集合【zoom,apple,Microsoft】
    查询bob,推荐1个,zoom

对于每个2请求,返回推荐的个数,注意关联集合有传递性

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值