Submission #1185379

#	Time	Username	Problem	Language	Result	Execution time	Memory
1185379		ansori	Matching (CEOI11_mat)	C++20	36 / 100	2096 ms	11332 KiB

mat

#include<bits/stdc++.h>
#pragma GCC optimize("O3")
#pragma GCC optimize("my solution")
//#define int long long
#define ll long long
#define nlp nullptr
#define btc __builtin_popcount
#define sht int_fast16_t
#define ld double
#define fi first
#define se second
using namespace std;
const int N = 3e5 + 5 , K = 8 , B = 369;
const int md = 998244353;
const int inf = 1e9;
const long double eps = 1e-8;
const bool stress = false;
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
bool chk(vector<int> &p , vector<int> &t , int ps , int sf){
	if(sf >= p.size()) return false;
	for(int i = 0;i < sf; ++ i){
		if((p[sf] > p[i] and t[ps] < t[ps - (sf - i)]) or (p[sf] < p[i] and t[ps] > t[ps - (sf - i)])) return false;
	}
	return true;
}
vector<int> kmp(vector<int> pat , vector<int> txt){
	int n = pat.size() , m = txt.size() , suf[n + 1] = {} , sf = 0;
	vector<int> ans;
	// for(int i = 0;i < n; ++ i) cout << pat[i] << ' ';
	// cout << '\n';
	for(int i = 1;i < n; ++ i){
		while(sf and ! chk(pat , pat , i , sf)) sf = suf[sf - 1];
		if(chk(pat , pat , i , sf)) sf ++; 
		suf[i] = sf;
		//cout << sf << ' ';
	}
	sf = 0;
	for(int i = 0;i < m; ++ i){
		while(sf and ! chk(pat , txt , i , sf)) sf = suf[sf - 1];
		if(chk(pat , txt , i , sf)) sf ++;
		if(sf == n) ans.push_back(i - n + 1);	
	}
	return ans;	
}
void solve(){
	int n , m;
	cin >> n >> m;
	vector<int> pat(n) , txt(m);
	for(int i = 0;i < n; ++ i){
		int s;
		cin >> s;
		pat[s - 1] = i;
	}
	for(int i = 0;i < m; ++ i) cin >> txt[i];
	vector<int> ans = kmp(pat , txt);
	cout << ans.size() << '\n';
	for(auto to : ans) cout << to + 1 << ' ';
}
main()
{
	//217
	//freopen("seating.in", "r", stdin) , freopen("seating.out" , "w", stdout);
	ios_base::sync_with_stdio(0);
	cin.tie(0),cout.tie(0);
    int T = 1;
    //cin >> T;
    while(T --)
    {
    	solve();
    	cout << '\n';
    }
}

Compilation message (stderr)

mat.cpp:3:35: warning: bad option '-fmy solution' to pragma 'optimize' [-Wpragmas]
    3 | #pragma GCC optimize("my solution")
      |                                   ^
mat.cpp:19:59: warning: bad option '-fmy solution' to attribute 'optimize' [-Wattributes]
   19 | bool chk(vector<int> &p , vector<int> &t , int ps , int sf){
      |                                                           ^
mat.cpp:26:50: warning: bad option '-fmy solution' to attribute 'optimize' [-Wattributes]
   26 | vector<int> kmp(vector<int> pat , vector<int> txt){
      |                                                  ^
mat.cpp:45:12: warning: bad option '-fmy solution' to attribute 'optimize' [-Wattributes]
   45 | void solve(){
      |            ^
mat.cpp:59:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   59 | main()
      | ^~~~
mat.cpp:59:6: warning: bad option '-fmy solution' to attribute 'optimize' [-Wattributes]
   59 | main()
      |      ^

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