add: 220513 [cpp]

This commit is contained in:
eat-swap 2022-05-13 20:56:43 +08:00
parent 326523b170
commit 523d79e94a
Signed by: Eatswap
GPG Key ID: BE661106A1F3FA0B
2 changed files with 54 additions and 1 deletions

53
cpp/2205/220513.cpp Normal file
View File

@ -0,0 +1,53 @@
#include <cstdio>
class Node {
public:
int val;
Node* left;
Node* right;
Node* next;
explicit Node(int _val = 0, Node* _left = nullptr, Node* _right = nullptr, Node* _next = nullptr) : val(_val), left(_left), right(_right), next(_next) {}
};
class Solution {
public:
static Node* connect(Node* root) {
if (!root)
return nullptr;
Node* qNode[6144] = {root};
int qLevel[6144] = {0};
int front = 0, rear = 1;
while (rear > front) {
Node* n = qNode[front];
int level = qLevel[front];
++front;
if (rear > front && level == qLevel[front]) {
n->next = qNode[front];
}
if (n->left) {
qNode[rear] = n->left;
qLevel[rear] = 1 + level;
++rear;
}
if (n->right) {
qNode[rear] = n->right;
qLevel[rear] = 1 + level;
++rear;
}
}
return root;
}
};
int main() {
auto* r = new Node(1, new Node(2, new Node(4), new Node(5)), new Node(3, nullptr, new Node(7)));
auto h = Solution::connect(r);
return 0;
}

View File

@ -3,4 +3,4 @@ PROJECT(2205)
SET(CMAKE_CXX_STANDARD 23)
ADD_EXECUTABLE(2205 220512.cpp)
ADD_EXECUTABLE(2205 220513.cpp)