add prime number problem and solution

James Lee · James Lee · commit 1cc9103ba483 · 2016-12-26T05:11:53.000Z
diff --git a/in/prime_nums.text b/in/prime_nums.text
@@ -0,0 +1,10 @@
+  2	  3	  5	  7	 11	 13	 17	 19	 23	 29
+ 31	 37	 41	 43	 47	 53	 59	 61	 67	 71
+ 73	 79	 83	 89	 97	101	103	107	109	113
+127	131	137	139	149	151	157	163	167	173
+179	181	191	193	197	199	211	223	227	229
+233	239	241	251	257	263	269	271	277	281
+283	293	307	311	313	317	331	337	347	349
+353	359	367	373	379	383	389	397	401	409
+419	421	431	433	439	443	449	457	461	463
+467	479	487	491	499	503	509	521	523	541
diff --git a/src/main/java/com/sparkTutorial/rdd/airports/AirportsProblem.java b/src/main/java/com/sparkTutorial/rdd/airports/AirportsProblem.java
@@ -6,7 +6,7 @@ public static void main(String[] args) throws Exception {
 
         /* TODO: Create a Spark program to read the airport data from in/airports.text, find all the airports which are located in United States and output the airport's name and the city's name to out/airports.text.
 
-           Each entry of the input file contains the following columns:
+           Each row of the input file contains the following columns:
 
            Airport ID, Name of airport, Main city served by airport, Country where airport is located, IATA/FAA code, ICAO Code, Latitude, Longitude, Altitude, Timezone, DST, Timezone in Olson format
 
diff --git a/src/main/java/com/sparkTutorial/rdd/sumOfPrimeNums/SumOfFirst100PrimeNumbersProblem.java b/src/main/java/com/sparkTutorial/rdd/sumOfPrimeNums/SumOfFirst100PrimeNumbersProblem.java
@@ -0,0 +1,12 @@
+package com.sparkTutorial.rdd.sumOfPrimeNums;
+
+public class SumOfFirst100PrimeNumbersProblem {
+
+    public static void main(String[] args) throws Exception {
+
+        /* TODO: Create a Spark program to read the first 100 prime numbers from in/prime_nums.text, print the sum of those numbers to console.
+
+           Each row of the input file contains 10 prime numbers separated by spaces.
+         */
+    }
+}
diff --git a/src/main/java/com/sparkTutorial/rdd/sumOfPrimeNums/SumOfFirst100PrimeNumbersSolution.java b/src/main/java/com/sparkTutorial/rdd/sumOfPrimeNums/SumOfFirst100PrimeNumbersSolution.java
@@ -0,0 +1,27 @@
+package com.sparkTutorial.rdd.sumOfPrimeNums;
+
+import org.apache.spark.SparkConf;
+import org.apache.spark.api.java.JavaRDD;
+import org.apache.spark.api.java.JavaSparkContext;
+
+import java.util.Arrays;
+
+public class SumOfFirst100PrimeNumbersSolution {
+
+    public static void main(String[] args) throws Exception {
+
+        SparkConf conf = new SparkConf().setAppName("primeNumbers").setMaster("local[*]");
+
+        JavaSparkContext sc = new JavaSparkContext(conf);
+
+        JavaRDD<String> lines = sc.textFile("in/prime_nums.text");
+
+        JavaRDD<String> numbers = lines.flatMap(line -> Arrays.asList(line.split("\\s+")).iterator());
+
+        JavaRDD<String> validNumbers = numbers.filter(number -> !number.equals(""));
+
+        String sum = validNumbers.reduce((x, y) -> Integer.valueOf(x) + Integer.valueOf(y) + "");
+
+        System.out.println("Sum is: " + sum);
+    }
+}
