user can specify the prefType through option

Author: nojima

main.cpp

@@ -1,15 +1,19 @@
 #include <cstdio>
+#include <cstdlib>
 #include <vector>
 #include "ap.h"
 using namespace std;
 
 // test
-int main()
+int main(int argc, char** argv)
 {
-  vector<int> examplar = affinityPropagation(stdin);
-  printf("examplar:");
+  int prefType = 1;
+  if (argc >= 2) {
+    prefType = atoi(argv[1]);
+  }
+  vector<int> examplar = affinityPropagation(stdin, prefType);
   for (size_t i = 0; i < examplar.size(); ++i) {
-    printf(" %d", examplar[i]);
+    printf("%d ", examplar[i]);
   }
   puts("");
   return 0;