Submission #879306

#	Time	Username	Problem	Language	Result	Execution time	Memory
879306		serifefedartar	Matching (CEOI11_mat)	C++17	100 / 100	864 ms	63684 KiB

mat

#include <bits/stdc++.h>
using namespace std;

#define fast ios::sync_with_stdio(0);cin.tie(0);
#define s second
#define f first
typedef long long ll;
const ll MOD = 1e9 + 9;
const ll LOGN = 21;
const ll MAXN = 1e6 + 100;

struct Node {
	int hash = 0, cnt = 0;
	Node() : hash(0), cnt(0) { }
};

const ll P = 3412113;
vector<int> A, B, cc;
Node sg[4 * MAXN];
int powP[MAXN];

void update(int k, int a, int b, int pos, int val) {
	if (b < pos || a > pos)
		return ;
	if (a == b) {
		sg[k].hash = val;
		sg[k].cnt = (val != 0);
		return ;
	}
	update(2*k, a, (a+b)/2, pos, val);
	update(2*k+1, (a+b)/2+1, b, pos, val);
	sg[k].cnt = sg[2*k].cnt + sg[2*k+1].cnt;
	sg[k].hash = (1LL * sg[2*k].hash * powP[sg[2*k+1].cnt] + 1LL * sg[2*k+1].hash) % MOD;
}

int main() {
	fast
	powP[0] = 1;
	for (int i = 1; i < MAXN; i++)
		powP[i] = (powP[i-1] * P) % MOD;
	int N, M;
	cin >> N >> M;

	A = vector<int>(N + 1);
	B = vector<int>(M + 1);
	for (int i = 1; i <= N; i++)
		cin >> A[i];

	ll main_hash = 0;
	for (int i = 1; i <= N; i++)
		main_hash = (main_hash * P + A[i]) % MOD;

	for (int i = 1; i <= M; i++) {
		cin >> B[i];
		cc.push_back(B[i]);
	}
	sort(cc.begin(), cc.end());

	for (int i = 1; i <= M; i++)
		B[i] = upper_bound(cc.begin(), cc.end(), B[i]) - cc.begin();

	ll sum = 0;
	for (int i = 0; i < N; i++)
		sum = (sum + powP[i]) % MOD;

	for (int i = 1; i < N; i++)
		update(1, 1, M + 10, B[i], i);

	vector<int> ans;
	for (int i = N; i <= M; i++) {
		update(1, 1, M + 10, B[i], i);

		ll hash_here = (sg[1].hash - (i - N) * sum % MOD + MOD) % MOD;
		if (hash_here == main_hash)
			ans.push_back(i - N + 1);

		update(1, 1, M + 10, B[i-N+1], 0); 
	}

	cout << ans.size() << "\n";
	for (auto u : ans)
		cout << u << " ";
	cout << "\n";
}

• Subtask 1: 9 / 9
• Subtask 2: 9 / 9
• Subtask 3: 9 / 9
• Subtask 4: 9 / 9
• Subtask 5: 9 / 9
• Subtask 6: 9 / 9
• Subtask 7: 9 / 9
• Subtask 8: 9 / 9
• Subtask 9: 8 / 8
• Subtask 10: 10 / 10
• Subtask 11: 10 / 10