modify

guide/basic.cc

@@ -5,11 +5,15 @@
  *
  * \author Tianqi Chen
  */
+#include <vector>
 #include <rabit.h>
 using namespace rabit;
-const int N = 3;
 int main(int argc, char *argv[]) {
-  int a[N];
+  int N = 3;
+  if (argc > 1) {
+    N = atoi(argv[1]);
+  }
+  std::vector<int> a(N);
   rabit::Init(argc, argv);
   for (int i = 0; i < N; ++i) {
     a[i] = rabit::GetRank() + i;