Initial commit

1 files changed, 24 insertions, 0 deletions

diff --git a/01_introductory_problems/two_knights_1072.cpp b/01_introductory_problems/two_knights_1072.cpp
new file mode 100644
index 0000000..15d5f55
--- /dev/null
+++ b/01_introductory_problems/two_knights_1072.cpp
@@ -0,0 +1,24 @@
+#include <iostream>
+#include <vector>
+
+int main() {
+        long long n, a;
+        std::vector<long long> b = {0, 0, 6, 28, 96};
+        std::cin >> n;
+
+        for (long long k = 1; k <= n; k++) {
+                if (k <= 4) {
+                        a = b[k];
+                } else {
+                        // Both knights in new strip
+                        a += (2*k - 1)*(k - 1) - 2;
+
+                        // One knight in new strip, one in previous square
+                        const long long s = (k-1)*(k-1);
+                        a += 5*(s - 2);
+                        a += 4*(s - 3);
+                        a += (2*k - 10)*(s - 4);
+                }
+                std::cout << a << std::endl;
+        }
+}