题意:给你一个字符串,'T'代表2, 'W'代表1。\(m\)次询问,每次问你有没有一个区间和等于\(x\),有则输出一个区间,否则输出"NIE"。
我们观察只给1和2这两个值有什么用,如果我们知道\(x\)是有的,并且区间为\(l_x\) 和 \(r_x\),那么如果\(s[l_x]\) 或者 \(s[r_x]\)为2,是不是能推出\(x-2\),否则两边都是1,也能推出\(x-2\),所有如果\(x\)存在,那么小于\(x\)的且和\(x\)奇偶性相同的数也存在。
那么我们找到最大的奇数和偶数就行了,先求\(sum_n\),得到一个最大的数,然后左右找一个奇数区间,减去后就得到另一个最大的数。然后从大到小循环确定方案就行了。
点击查看代码
void solve() {int n, m;std::cin >> n >> m;std::string s;std::cin >> s;std::vector<int> sum(n + 1);for (int i = 0; i < n; ++ i) {sum[i + 1] = sum[i] + (s[i] == 'T' ? 2 : 1);}std::vector<int> l(2 * n + 1, n + 1), r(2 * n + 1);l[sum[n]] = 1; r[sum[n]] = n;for (int i = 1; i <= n; ++ i) {if (sum[i] & 1) {l[sum[n] - sum[i]] = i + 1;r[sum[n] - sum[i]] = n;break;}}for (int i = n; i >= 1; -- i) {if ((sum[n] - sum[i - 1]) & 1) {l[sum[i - 1]] = 1;r[sum[i - 1]] = i - 1;}}for (int i = 2 * n; i >= 2; -- i) {if (l[i] <= r[i]) {if (s[l[i] - 1] == 'T') {l[i - 2] = l[i] + 1;r[i - 2] = r[i];} else if (s[r[i] - 1] == 'T') {l[i - 2] = l[i];r[i - 2] = r[i] - 1;} else {l[i - 2] = l[i] + 1;r[i - 2] = r[i] - 1;}}}while (m -- ) {int x;std::cin >> x;if (x <= 2 * n && l[x] <= r[x]) {std::cout << l[x] << " " << r[x] << "\n";} else {std::cout << "NIE\n";}}
}