题解 | #手串#

手串

https://www.nowcoder.com/practice/0bb1fad52f474bdaa4d7636ca3a98244

思路1:最直观的想法是,不用存储每个串珠对应的颜色数组,直接存储每种颜色出现的串珠位置,使用colors存储每种颜色出现的串珠位置!遍历每种颜色并判断连续m个是否出现重复:首先判断首尾出现的位置差,由于是按照顺序填充数组,故colors[i]有序,那么可以使用(start+n-end)<m进行判断;接着判断每个相邻位置差,一旦发现连续m个出现重复,就直接判断下一种颜色。

#include<bits/stdc++.h>
using namespace std;
int main() {
    int n, m, c;
    int ans = 0;
    cin >> n >> m >> c;
    //存储颜色出现的串珠位置
    vector<int> colors[c + 1];
    int num, color;
    for (int i = 0; i < n; i++) {
        cin >> num;
        for (int j = 0; j < num; j++) {
            cin >> color;
            colors[color].push_back(i);
        }
    }
    // for(int i=1;i<=c;i++)
    // {
    //   for(int j=0;j<colors[i].size();j++)
    //   {
    //     cout<<colors[i][j]<<" ";
    //   }
    //   cout<<endl;
    // }
    //判断连续m个是否出现重复
    //只需对每个颜色判断同颜色相邻的位置差是否小于m
    for (int i = 1; i <= c; i++) {
        //先判断首尾出现的位置差
        if (colors[i][0] + n - colors[i][colors[i].size() - 1] < m) {
            ans++;
            continue;
        }
        //判断每个相邻位置差
        for (int j = 0; j < colors[i].size() - 1; j++) {
            if (colors[i][j + 1] - colors[i][j] < m) {
                ans++;
                break;
            }
        }
    }
    cout << ans;
    return 0;
}

总结:注意,如果直接vector<vector<int>> colors(n),那么会生成n乘以n的二维数组喔!建议使用vector<int> colors[n],那么即是n个vector<int>。

#手串#
字节跳动前端 文章被收录于专栏

字节跳动公司真题!!我最爱字节跳动!!

全部评论

相关推荐

11-08 13:58
门头沟学院 Java
程序员小白条:竟然是蓝桥杯人才doge,还要花钱申领的offer,这么好的公司哪里去找
点赞 评论 收藏
分享
jack_miller:杜:你不用我那你就用我的美赞臣
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务