leetcode::shortest_distance_to_target_color::Solution类 参考

#include <leetcode.h>

静态 Public 成员函数
static vector< int >	shortestDistanceColor (vector< int > &colors, vector< vector< int > > &queries)

详细描述

在文件 leetcode.h 第 2761 行定义.

成员函数说明

shortestDistanceColor()

vector< int > leetcode::shortest_distance_to_target_color::Solution::shortestDistanceColor ( vector< int > & colors, vector< vector< int > > & queries )

static

在文件 leetcode.cpp 第 7572 行定义.

                                                                                                     {
            vector<int> ans;
            unordered_map<int, vector<int>> um;
            for(int i = 0; i < colors.size(); i++) {
                um[colors[i]].emplace_back(i);
            }
            for(auto &[k, vec]: um) {
                sort(vec.begin(), vec.end());
            }
            for(auto &query: queries) {
                int &i           = query[0];
                int &c           = query[1];
                vector<int> &vec = um[c];
                auto g           = lower_bound(vec.begin(), vec.end(), i, less());
                auto l           = lower_bound(vec.rbegin(), vec.rend(), i, greater());
                if(g == vec.end() && l == vec.rend()) {
                    ans.emplace_back(-1);
                } else if(g == vec.end()) {
                    ans.emplace_back(abs(*l - i));
                } else if(l == vec.rend()) {
                    ans.emplace_back(abs(*g - i));
                } else {
                    ans.emplace_back(min(abs(*l - i), abs(*g - i)));
                }
            }
            return ans;
        }

被这些函数引用 leetcode::shortest_distance_to_target_color::TEST() , 以及 leetcode::shortest_distance_to_target_color::TEST().

---

该类的文档由以下文件生成:

• leetcode.h
• leetcode.cpp