Dodano rozwiązanie zadania 'Trzy wieże' z XXII OI

Author: Kulezi

rozwiazania/xxii/etap2/dzien2/trz/testuj.sh

@@ -0,0 +1,44 @@
+#!/bin/bash
+
+compile() {
+    g++ -O3 -static -g $1.cpp -o $1
+}
+
+MODEL=$1
+BRUTE=$2
+GEN=$3
+
+trap 'echo STOP; exit 0' INT
+
+run() {
+    /usr/bin/time -f "seconds=%e\nkbytes=%M\nexitcode=%x" -o $1.time ./$1 < test.in > $1.out 2> $1.err
+}
+
+info() {
+    source $1.time
+    echo -ne "(${1}: ${seconds}s, $((kbytes / 1024)) MB) "
+}
+
+set -x
+compile $MODEL && compile $BRUTE && compile $GEN
+set +x
+
+for i in $(seq 1 100000); do
+    echo -n $i
+
+    ./$GEN > test.in
+    run $1 
+    run $2
+    
+    cmp $1.out $2.out
+    if [[ $? -ne 0 ]]; then
+        echo -ne " [FAIL] "
+        exit 1
+    fi
+
+    echo -n " [OK] "
+    info $1
+    info $2
+    echo ""
+
+done

rozwiazania/xxii/etap2/dzien2/trz/trz.cpp

@@ -0,0 +1,67 @@
+// Rozwiązanie zadania 'Trzy wieże' z II etapu XXII OI.
+// Autor rozwiązania: Paweł Putra
+// Złożoność czasowa: O(n)
+// Złożoność pamięciowa: O(n)
+// Punkty: 100 (upsolve)
+
+#include <iostream>
+#include <cstdint>
+#include <string>
+#include <cassert>
+#define dbg(x) #x << " = " << x << " "
+using namespace std;
+int map(char c) {
+    if (c == 'C') return 0;
+    if (c == 'B') return 1;
+    if (c == 'S') return 2;
+    assert(false);
+    return -1;
+}
+constexpr int MAXN = 1'000'005, KOLORY = 3;
+int pref[MAXN][KOLORY];
+int suma(int l, int r, int tab) {
+    return pref[r][tab] - pref[l-1][tab];
+}
+
+bool rozne(int x, int y) {
+    return x != y || x == 0 || y == 0;
+}
+
+bool dobry(int l, int r) {
+    int a = suma(l, r, 0);
+    int b = suma(l, r, 1);
+    int c = suma(l, r, 2);
+    return rozne(a, b) && rozne(a, c) && rozne(b, c);  
+}
+
+int res;
+void maksuj(int i, int j) {
+    if (j - i + 1 > res && dobry(i, j)) {
+        res = j - i + 1;
+    }
+}
+
+int32_t main() {
+    ios_base::sync_with_stdio(0);
+    int n;
+    string s;
+    
+    cin >> n >> s;
+
+    for (int i = 1; i <= n; i++) {
+        for (int j = 0; j < KOLORY; j++) {
+            pref[i][j] = pref[i - 1][j];
+        }
+
+        pref[i][map(s[i-1])]++;
+    }
+
+    for (int i = 1; i <= 5; i++) {
+        for (int j = i; j <= n; j++) {
+            maksuj(i, j);
+            maksuj(n-j+1, n-i+1);
+        }
+    }
+
+    cout << res << "\n";
+}

rozwiazania/xxii/etap2/dzien2/trz/trz60.cpp

@@ -0,0 +1,62 @@
+// Rozwiązanie powolne zadania 'Trzy wieże' z II etapu XXII OI.
+// Autor rozwiązania: Paweł Putra
+// Złożoność czasowa: O(n^2)
+// Złożoność pamięciowa: O(n)
+// Punkty: 60 (upsolve)
+
+#include <iostream>
+#include <cstdint>
+#include <string>
+#include <cassert>
+#define dbg(x) #x << " = " << x << " "
+using namespace std;
+int map(char c) {
+    if (c == 'C') return 0;
+    if (c == 'B') return 1;
+    if (c == 'S') return 2;
+    assert(false);
+    return -1;
+}
+constexpr int MAXN = 1'000'005, KOLORY = 3;
+int pref[MAXN][KOLORY];
+int suma(int l, int r, int tab) {
+    return pref[r][tab] - pref[l-1][tab];
+}
+
+bool rozne(int x, int y) {
+    return x != y || x == 0 || y == 0;
+}
+
+bool dobry(int l, int r) {
+    int a = suma(l, r, 0);
+    int b = suma(l, r, 1);
+    int c = suma(l, r, 2);
+    return rozne(a, b) && rozne(a, c) && rozne(b, c);  
+}
+
+int32_t main() {
+    ios_base::sync_with_stdio(0);
+    int n;
+    string s;
+    
+    cin >> n >> s;
+
+    for (int i = 1; i <= n; i++) {
+        for (int j = 0; j < KOLORY; j++) {
+            pref[i][j] = pref[i - 1][j];
+        }
+
+        pref[i][map(s[i-1])]++;
+    }
+
+    int res = 0;
+    for (int i = 1; i <= n; i++) {
+        for (int j = i; j <= n; j++) {
+            if (j - i + 1 > res && dobry(i, j)) {
+                res = j - i + 1;
+            }
+        }
+    }
+
+    cout << res << "\n";
+}

rozwiazania/xxii/etap2/dzien2/trz/trzg.cpp

@@ -0,0 +1,22 @@
+// Generatorka do zadania 'Trzy wieże' z II etapu XXII OI.
+
+#include <random>
+#include <iostream>
+#include <chrono>
+using namespace std;
+mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
+int R(int l, int r) {
+    return uniform_int_distribution<>(l, r)(rng);
+}
+
+string alpha = "BSC";
+int main() {
+    int n = R(1, 16);
+    cout << n << endl;
+
+    for (int i = 1; i <= n; i++) {
+        cout << alpha[R(0, 2)];
+    }
+
+    cout << "\n";
+}