leetcodeSolution to some Leetcode problems written in C++ |
git clone git://git.dimitrijedobrota.com/leetcode.git |
Log | Files | Refs | README | LICENSE |
2672.cpp (845B)
0 class Solution {
1 public:
2 vector<int> colorTheArray(const int n, const vector<vector<int>> &queries) const {
3 static int nums[100002];
4 memset(nums, 0x00, sizeof(nums));
6 const int m = size(queries);
7 vector<int> res(m);
9 int crnt = 0;
10 for (int i = 0; i < m; i++) {
11 const int idx = queries[i][0] + 1;
12 const int color = queries[i][1];
13 if (nums[idx] != color) {
14 if (nums[idx]) {
15 if (nums[idx] == nums[idx - 1]) crnt--;
16 if (nums[idx] == nums[idx + 1]) crnt--;
17 }
18 nums[idx] = color;
19 if (nums[idx] == nums[idx - 1]) crnt++;
20 if (nums[idx] == nums[idx + 1]) crnt++;
21 }
22 res[i] = crnt;
23 }
25 return res;
26 }
27 };