Submission #115550

#	Time	Username	Problem	Language	Result	Execution time	Memory
115550		gs14004	Matching (CEOI11_mat)	C++17	100 / 100	646 ms	48820 KiB

mat

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <limits.h>
#include <math.h>
#include <time.h>
#include <iostream>
#include <functional>
#include <numeric>
#include <algorithm>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <string>
#include <bitset>
#include <map>
#include <set>
using namespace std;
typedef long long lint;
typedef long double llf;
typedef pair<int, int> pi;

int n, m, a[1000005], b[1000005];
int lp[1000005], rp[1000005], fail[1000005];

bool ins1(int x, int p){
	return a[x - lp[p]] <= a[x] && a[x] <= a[x - rp[p]];
}

bool ins2(int x, int p){
	return b[x - lp[p]] <= b[x] && b[x] <= b[x - rp[p]];
}

int nxtl[1000005], nxtr[1000005], r[1000005];

int fl(int x){
	if(x < 0) return x;
	return nxtl[x] = (nxtl[x] == x ? x : fl(nxtl[x]));
}

int fr(int x){
	if(x >= n) return x;
	return nxtr[x] = (nxtr[x] == x ? x : fr(nxtr[x]));
}

int main(){
	cin >> n >> m;
	for(int i=0; i<n; i++){
		int v;
		scanf("%d",&v);
		a[v-1] = i;
		r[i] = v-1;
		nxtl[i] = nxtr[i] = i;
	}
	for(int i=n-1; i>=0; i--){
		nxtl[a[i]] = fl(a[i] - 1);
		nxtr[a[i]] = fr(a[i] + 1);
		if(nxtr[a[i]] < n) rp[i] = i - r[nxtr[a[i]]];
		if(nxtl[a[i]] >= 0) lp[i] = i - r[nxtl[a[i]]];
	}
	int p = 0;
	for(int i=1; i<n; i++){
		while(p > 0 && !ins1(i, p)){
			p = fail[p];
		}
		if(ins1(i, p)) p++;
		fail[i+1] = p;
	}
	p = 0;
	vector<int> v;
	for(int i=0; i<m; i++){
		scanf("%d",&b[i]);
		while(p > 0 && !ins2(i, p)){
			p = fail[p];
		}
		if(ins2(i, p)) p++;
		if(p == n){
			v.push_back(i - n + 2);
			p = fail[p];
		}
	}
	printf("%d\n", v.size());
	for(auto &i : v) printf("%d ",i);
}

Compilation message (stderr)

mat.cpp: In function 'int main()':
mat.cpp:84:25: warning: format '%d' expects argument of type 'int', but argument 2 has type 'std::vector<int>::size_type {aka long unsigned int}' [-Wformat=]
  printf("%d\n", v.size());
                 ~~~~~~~~^
mat.cpp:52:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d",&v);
   ~~~~~^~~~~~~~~
mat.cpp:74:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
   scanf("%d",&b[i]);
   ~~~~~^~~~~~~~~~~~

• 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