Add const size for array

Author: Mohamed Ayman

src/lca_demo.cpp

@@ -1,10 +1,14 @@
-#include <bits/stdc++.h>
+#include <cstdio>
+#include <vector>
 
-int numberOfNodes, MAXLOG;
+const int MAX_NODE = 5000;
+const int MAX_LOG = 20;
+
+int numberOfNodes, maxLog;
 std::vector< std::vector<int> > adjList;
-int parent[50005], nodeHeight[50005];
-bool visited[50005];
-int binaryLiftDp[50005][27];
+int parent[MAX_NODE], nodeHeight[MAX_NODE];
+bool visited[MAX_NODE];
+int binaryLiftDp[MAX_NODE][MAX_LOG];
 
 void dfs(int currentNode, int currentParent)
 {
@@ -31,10 +35,10 @@ int getMaxLog(){
 void initializeDP()
 {
  nodeHeight[-1] = -1;
- MAXLOG = getMaxLog();
+ maxLog = getMaxLog();
  dfs(0, -1);
  for(int i = 0; i < numberOfNodes; i++) binaryLiftDp[i][0] = parent[i];
- for(int i = 1; i <= MAXLOG; i++)
+ for(int i = 1; i <= maxLog; i++)
  {
  for(int j = 0; j < numberOfNodes; j++)
  {
@@ -48,13 +52,13 @@ void initializeDP()
 int LCA(int a, int b)
 {
  if(nodeHeight[a] < nodeHeight[b]) std::swap(a,b);
- for(int i = MAXLOG; i >= 0; i--)
+ for(int i = maxLog; i >= 0; i--)
  {
  if(binaryLiftDp[a][i] + 1 && nodeHeight[binaryLiftDp[a][i]] >= nodeHeight[b])
  a = binaryLiftDp[a][i];
  }
  if(!(a - b)) return a;
- for(int i = MAXLOG; i >= 0; i--)
+ for(int i = maxLog; i >= 0; i--)
  {
  if(binaryLiftDp[a][i] + 1 && binaryLiftDp[a][i] - binaryLiftDp[b][i])
  a = binaryLiftDp[a][i], b = binaryLiftDp[b][i];