// 2022-1-wzt-wielki-zderzacz-termionow.cpp : This file contains the 'main' function. Program execution begins and ends there. // #include <iostream> constexpr long long MD = 1e9 + 7; int a, b, c, hm1, hm2; std::string s; constexpr int MAXN = 1e6; long long t[MAXN][3]; void chng(int i, char chr) { if (chr == s[i]) { return; } if (s[i] == 'N') { c--; } if (s[i] == 'C') { a--; } if (s[i] == 'Z') { b--; } if (chr == 'N') { c++; } if (chr == 'C') { a++; } if (chr == 'Z') { b++; } // a, b, c done // hm1 if (i % 2) { if (s[i] == 'Z') { hm1--; } if (chr == 'Z') { hm1++; } } else { if (s[i] == 'C') { hm1--; } if (chr == 'C') { hm1++; } } // hm2 if (i % 2 == 0) { if (s[i] == 'Z') { hm2--; } if (chr == 'Z') { hm2++; } } else { if (s[i] == 'C') { hm2--; } if (chr == 'C') { hm2++; } } s[i] = chr; } long long getres() { //std::cout << a << ' ' << b << ' ' << c << ' ' << hm1 << ' ' << hm2 << '\n'; long long res = 0; for (int r = 0; r < 3; r++) { if ((s.size() % 3) != ((2 * (a + r)) % 3)) { res += t[c][r]; res %= MD; } } if (s.size() % 2 == 1) { if (hm1 == 0) { res--; } if (hm2 == 0) { res--; } } res += MD; res %= MD; return res; } int main() { t[0][0] = 1; for (int i = 1; i < MAXN; i++) { for (int r = 0; r < 3; r++) { t[i][r] = (t[i - 1][r] + t[i - 1][(r + 2) % 3]) % MD; } } std::ios_base::sync_with_stdio(0); std::cin.tie(0); std::cout.tie(0); int n, q; char chr; std::cin >> n >> q; a = 0; b = 0; c = n; hm1 = 0; hm2 = 0; s = std::string(n, 'N'); for (int i = 0; i < n; i++) { std::cin >> chr; chng(i, chr); } std::cout << getres() << '\n'; int i; while (q--) { std::cin >> i >> chr; i--; chng(i, chr); std::cout << getres() << '\n'; } } // Run program: Ctrl + F5 or Debug > Start Without Debugging menu // Debug program: F5 or Debug > Start Debugging menu // Tips for Getting Started: // 1. Use the Solution Explorer window to add/manage files // 2. Use the Team Explorer window to connect to source control // 3. Use the Output window to see build output and other messages // 4. Use the Error List window to view errors // 5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project // 6. In the future, to open this project again, go to File > Open > Project and select the .sln file
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 | // 2022-1-wzt-wielki-zderzacz-termionow.cpp : This file contains the 'main' function. Program execution begins and ends there. // #include <iostream> constexpr long long MD = 1e9 + 7; int a, b, c, hm1, hm2; std::string s; constexpr int MAXN = 1e6; long long t[MAXN][3]; void chng(int i, char chr) { if (chr == s[i]) { return; } if (s[i] == 'N') { c--; } if (s[i] == 'C') { a--; } if (s[i] == 'Z') { b--; } if (chr == 'N') { c++; } if (chr == 'C') { a++; } if (chr == 'Z') { b++; } // a, b, c done // hm1 if (i % 2) { if (s[i] == 'Z') { hm1--; } if (chr == 'Z') { hm1++; } } else { if (s[i] == 'C') { hm1--; } if (chr == 'C') { hm1++; } } // hm2 if (i % 2 == 0) { if (s[i] == 'Z') { hm2--; } if (chr == 'Z') { hm2++; } } else { if (s[i] == 'C') { hm2--; } if (chr == 'C') { hm2++; } } s[i] = chr; } long long getres() { //std::cout << a << ' ' << b << ' ' << c << ' ' << hm1 << ' ' << hm2 << '\n'; long long res = 0; for (int r = 0; r < 3; r++) { if ((s.size() % 3) != ((2 * (a + r)) % 3)) { res += t[c][r]; res %= MD; } } if (s.size() % 2 == 1) { if (hm1 == 0) { res--; } if (hm2 == 0) { res--; } } res += MD; res %= MD; return res; } int main() { t[0][0] = 1; for (int i = 1; i < MAXN; i++) { for (int r = 0; r < 3; r++) { t[i][r] = (t[i - 1][r] + t[i - 1][(r + 2) % 3]) % MD; } } std::ios_base::sync_with_stdio(0); std::cin.tie(0); std::cout.tie(0); int n, q; char chr; std::cin >> n >> q; a = 0; b = 0; c = n; hm1 = 0; hm2 = 0; s = std::string(n, 'N'); for (int i = 0; i < n; i++) { std::cin >> chr; chng(i, chr); } std::cout << getres() << '\n'; int i; while (q--) { std::cin >> i >> chr; i--; chng(i, chr); std::cout << getres() << '\n'; } } // Run program: Ctrl + F5 or Debug > Start Without Debugging menu // Debug program: F5 or Debug > Start Debugging menu // Tips for Getting Started: // 1. Use the Solution Explorer window to add/manage files // 2. Use the Team Explorer window to connect to source control // 3. Use the Output window to see build output and other messages // 4. Use the Error List window to view errors // 5. Go to Project > Add New Item to create new code files, or Project > Add Existing Item to add existing code files to the project // 6. In the future, to open this project again, go to File > Open > Project and select the .sln file |