萌新求助 8 分 /cf
查看原帖
萌新求助 8 分 /cf
852144
Loser_Syx楼主2024/7/18 08:56

Z 函数贺的 oi-wiki 所以是对的,匹配就挂了。

#include <iostream>
#include <cassert>
#include <queue>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <bitset>
#include <random>
#include <ctime>
#include <unordered_map>
#include <set>
using namespace std;
#define int long long
#define pii pair<int, int>
#define eb emplace_back
#define F first
#define S second
#define test(x) cout << "Test: " << x << '\n'
#define lowbit(x) (x & -x)
#define debug puts("qwq");
#define open(x) freopen(#x".in", "r", stdin);freopen(#x".out", "w", stdout);
#define close fclose(stdin);fclose(stdout);
namespace FastIO {
	template <typename T = int>
	inline T read() {
		T s = 0, w = 1;
		char c = getchar();
		while (!isdigit(c)) {
			if (c == '-') w = -1;
			c = getchar();
		}
		while (isdigit(c)) s = (s << 1) + (s << 3) + (c ^ 48), c = getchar();
		return s * w;
	}
	template <typename T>
	inline void read(T &s) {
		s = 0;
		int w = 1;
		char c = getchar();
		while (!isdigit(c)) {
			if (c == '-') w = -1;
			c = getchar();
		}
		while (isdigit(c)) s = (s << 1) + (s << 3) + (c ^ 48), c = getchar();
		s = s * w;
	}
	template <typename T, typename... Arp> inline void read(T &x, Arp &...arp) {
		read(x), read(arp...);
	}
	template <typename T>
	inline void write(T x, char ch = '\n') {
		if (x < 0) x = -x, putchar('-');
		static char stk[25];
		int top = 0;
		do {
			stk[top++] = x % 10 + '0', x /= 10;
		} while (x);
		while (top) putchar(stk[--top]);
		putchar(ch);
		return;
	}
	template <typename T>
	inline void smax(T &x, T y) {
		if (x < y) x = y;
	}
	template <typename T>
	inline void smin(T &x, T y) {
		if (x > y) x = y;
	}
	void quit() {
		exit(0);
	}
} using namespace FastIO;
vector<int> z_func(string s) {
	int n = (int)s.length();
	vector<int> z(n, 0);
	for (int i = 1, l = 0, r = 0; i < n; ++i) {
		if (i <= r && z[i - l] < r - i + 1) {
			z[i] = z[i - l];
		} else {
			z[i] = max(0ll, r - i + 1);
			while (i + z[i] < n && s[z[i]] == s[i + z[i]]) ++z[i];
		}
		if (i + z[i] - 1 > r) l = i, r = i + z[i] - 1;
	}
	z[0] = n;
	return z;
}
signed main() {
	string s, b;
	cin >> b >> s;
	vector<int> z = z_func(s);
	int ans1 = 0;
	for (int i = 0; i < (int)z.size(); ++i) ans1 ^= (z[i] + 1) * (i + 1);
	write(ans1);
	vector<int> nxt(b.size(), 0);
	int n = b.size(), m = s.size();
	for (int i = 0, l = 0, r = 0; i < n; ++i) {
		if (i <= r && z[i - l] < r - i + 1) {
			nxt[i] = z[i - l];
		} else {
			nxt[i] = max(0ll, r - i + 1);
			while (i + nxt[i] < n && b[i + nxt[i]] == s[nxt[i] + 1]) ++nxt[i];
		}
		if (i + nxt[i] - 1 > r) l = i, r = i + nxt[i] - 1;
	} ans1 = 0;
	for (int i = 0; i < n; ++i) ans1 ^= (nxt[i] + 1) * (i + 1);
	write(ans1);
	return 0;
}
2024/7/18 08:56
加载中...