1. LeetCode单周赛第320场
1.1 数组中不等三元组的数目
1.1.1 原题链接：力扣
https://leetcode.cn/problems/number-of-unequal-triplets-in-array/
1.1.2 解题思路：
暴力遍历咯。
1.1.3 代码：
class Solution {
public:
int unequalTriplets(vector<int>& nums) {
int n = nums.size();
int res = 0;
for(int i = 0; i < n; i ++ )
for(int j = i + 1; j < n; j ++ )
for(int k = j + 1; k < n; k ++ )
if(nums[i] != nums[j] && nums[i] != nums[k] && nums[j] != nums[k])
res ++;
return res;
}
};1.2 二叉搜索树最近节点查询
1.2.1 原题链接：力扣https://leetcode.cn/problems/closest-nodes-queries-in-a-binary-search-tree/
1.2.2 解题思路：
1、先把树种的元素存到数组中；
2、把数组中的元素排序，然后二分找满足要求的元素。
1.2.3 代码：
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
void tra(TreeNode* root, vector<int>& t)
{
if(root == nullptr) return;
t.push_back(root->val);
tra(root->left, t);
tra(root->right, t);
}
vector<vector<int>> closestNodes(TreeNode* root, vector<int>& queries) {
vector<vector<int>> res;
vector<int> tmp;
tra(root, tmp);
sort(tmp.begin(), tmp.end());
int n = tmp.size();
for(auto& x: queries) {
int a, b;
int l = 0, r = n - 1;
while(l < r) {
int mid = l + r + 1 >> 1;
if(tmp[mid] > x) r = mid - 1;
else l = mid;
}
if(tmp[l] <= x) a = tmp[l];
else a = -1;
l = 0, r = n - 1;
while(l < r) {
int mid = l + r >> 1;
if(tmp[mid] < x) l = mid + 1;
else r = mid;
}
if(tmp[l] >= x) b = tmp[l];
else b = -1;
res.push_back({a, b});
}
return res;
}
};1.3 到达首都最少油耗
1.3.1 原题链接：力扣https://leetcode.cn/problems/minimum-fuel-cost-to-report-to-the-capital/
1.3.2 解题思路：
以 0 为根，统计每个子树中节点的个数cur，那么每个子树需要的车的数量就是 cur / seats 向上取整。
1.3.3 代码：
#define N 100010
#define M 200010
int h[N], e[M], ne[M], idx;
typedef long long ll;
class Solution {
public:
ll ans = 0;
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
ll dfs(int u, int father, int seats)
{
ll cur = 1;
for(int i = h[u]; i != -1; i = ne[i]){
int j = e[i];
if(j != father) {
cur += dfs(j, u, seats);
}
}
if(u != 0) {
ans += (cur + seats - 1) / seats; //向上取整
}
return cur;
}
long long minimumFuelCost(vector<vector<int>>& roads, int seats) {
memset(h, -1, sizeof h);
idx = 0;
for(auto& road: roads) {
add(road[0], road[1]);
add(road[1], road[0]);
}
dfs(0, -1, seats);
return ans;
}
};2.AcWing周赛第78场
2.1 商品种类
2.1.1 原题链接：4719. 商品种类 - AcWing题库
2.1.2 解题思路：
使用set将产品的产地+名称存储到set中，最后返回set的大小即可。
2.1.3 代码：
#include <iostream>
#include <cstring>
#include <algorithm>
#include <unordered_set>
using namespace std;
int main()
{
int n;
cin >> n;
unordered_set<string> st;
for(int i = 0; i < n; i ++ ) {
string a, b;
cin >> a >> b;
st.insert(a + ' ' + b);
}
cout << st.size() << endl;
return 0;
}2.2 字符串
2.2.1 原题链接：4720. 字符串 - AcWing题库
2.2.2 解题思路：
把字符串 s 中的字符一个一个存到答案字符串 res 中，若与 res 中的最后一个字符相同，则把字符串 res 中的最后一个字符删去，反之，则加入到 res 中。最后输出 res 即可。
2.2.3 代码：
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
int main()
{
string s;
cin >> s;
string res;
for(auto c: s) {
if(res.size() && res.back() == c) res.pop_back();
else res += c;
}
cout << res << endl;
return 0;
}2.3 排队
2.3.1 原题链接：4721. 排队 - AcWing题库
2.3.2 解题思路：
1、用单调队列+二分的思路解决此问题；
2、从后往前遍历身高，栈里存的是对应身高的下标，每次对栈内下标二分找到比自身矮的最右侧一个人的位置；
3、对于栈内元素的维护，具有单调性，因为例如，对于寻找比 a[k] （k > i && k > j）的矮的最右侧的一位在哪，如果a[i], a[j]均小于a[k],且 i < j，那么，a[i]就没有存在必要了。
2.3.3 代码：
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
int h[N], stk[N];
int n;
int ans[N];
int main()
{
scanf("%d", &n);
for(int i = 0; i < n; i ++ ) scanf("%d", &h[i]);
int top = 0;
for(int i = n - 1; i >= 0; i -- ) {
if(!top || h[i] <= h[stk[top]]) ans[i] = -1;
else {
int l = 1, r = top;
while(l < r) {
int mid = l + r >> 1;
if(h[stk[mid]] < h[i]) r = mid;
else l = mid + 1;
}
ans[i] = stk[r] - i - 1;
}
if(!top || h[i] < h[stk[top]]) stk[++ top] = i;
}
for(int i = 0; i < n; i ++ )
printf("%d ", ans[i]);
return 0;
}
