1 files changed, 28 insertions, 0 deletions
diff --git a/ee1301/wk6/lab5/momentum.cpp b/ee1301/wk6/lab5/momentum.cpp
new file mode 100644
index 0000000..dfbaeee
--- /dev/null
+++ b/ee1301/wk6/lab5/momentum.cpp
@@ -0,0 +1,28 @@
+#include <iostream>
+
+double momentum (double velocity, double mass);
+
+int main() {
+  double velocity[3];
+  double mass;
+  std::cout << "Please enter velocity (x y z) [m/s]: ";
+  std::cin >> velocity[0] >> velocity[1] >> velocity [2];
+  std::cout << "Please enter mass [kg]: ";
+  std::cin >> mass;
+  
+  double* vector;
+  vector = new double[3];
+  std::cout << "Momentum: <";
+  for (int i=0; i<=2; i++) {
+    vector[i]=momentum(velocity[i], mass);
+    std::cout << vector[i];
+    if ((i<2))
+      std::cout << ",";
+  }
+  delete[] vector;
+  std::cout << ">" << std::endl;
+}
+
+double momentum(double velocity, double mass) {
+  return velocity * mass;
+}
+\ No newline at end of file