제출 #463420

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
463420		amunduzbaev	Exam (eJOI20_exam)	C++14	77 / 100	188 ms	1400 KiB

exam

#include "bits/stdc++.h"
using namespace std;
 
//~ #define int long long
 
void solve1(int n, vector<int>& a, vector<int>& b){
	vector<int> dp(n);
	if(n <= 5005){
		int res = 0;
		for(int i=0;i<n;i++){
			int s = i, e = i;
			while(s + 1 < n && a[s + 1] <= a[i]) s++;
			while(e && a[e - 1] <= a[i]) e--;

			vector<int> cnt(n);
			for(int j=0;j<n;j++){
				if(j) cnt[j] = cnt[j-1];
				cnt[j] += (b[j] == a[i]);
			}

			vector<int> pref(n, -1e9);
			for(int j=e;j<n;j++){
				if(j) pref[j] = pref[j-1];
				pref[j] = max(pref[j], (j ? - cnt[j-1] + dp[j-1] : 0));
			}

			for(int r=s;r>=e;r--){
				//~ int mx = -1e9;
				//~ for(int l=r;l>=e;l--){
					//~ mx = max((l ? - cnt[l-1] + dp[l-1] : 0), mx);
					//~ mx = max(mx, (l ? - cnt[l-1] + dp[l-1] : 0));
				//~ }
				
				dp[r] = max(dp[r], cnt[r] + pref[r]);
			}
		}
		
		for(int i=0;i<n;i++) res = max(res, dp[i]);
		cout<<res<<"\n";
	} else {
		int res;
		for(int i=0;i<n;i++){
			int r = i, l = i;
			while(r + 1 < n && a[r + 1] <= a[i]) r++;
			while(l > 0 && a[l - 1] <= a[l]) l--;
			vector<int> cnt(n), pref(n);
			for(int j=0;j<n;j++){
				if(j) cnt[j] = cnt[j-1];
				cnt[j] += (b[j] == a[i]);
			}
 
			for(int i=l;i<=r;i++){
				if(i) pref[i] = pref[i-1];
				pref[i] = max(pref[i], (i ? dp[i - 1] - cnt[i - 1] : 0));
			}
 
			for(int j=l;j<=r;j++){
				dp[j] = max(dp[j], cnt[j] + pref[j]);
			}
		}
 
		for(int i=0;i<n;i++) res = max(res, dp[i]);
		cout<<res<<"\n";
	}
}
 
void solve2(int n, vector<int>& a, vector<int>& b){
	int last = 0, cnt = 0, res = 0;
	for(int i=0;i<n;i++){
		if(a[i] > b[0]){
			if(cnt){
				res += (i - last);
			}
			
			cnt = 0, last = i + 1;
		} else {
			cnt |= (a[i] == b[i]);
		}
	} if(cnt){
		res += (n - last);
	}
	
	cout<<res<<"\n";
}
 
void solve(){
	int n; cin>>n;
	vector<int> a(n), b(n);
	for(auto& x : a) cin>>x;
	for(auto& x : b) cin>>x;
	
	int mx = 0, mn = 1e9;
	for(auto x : b) mx = max(mx, x), mn = min(mn, x);
	if(mx == mn){
		solve2(n, a, b);
		return;
	}
	
	if(n <= 5005){
		solve1(n, a, b);
	} else {
		solve2(n, a, b);
	}
}
 
/*
 
3
1 2 3
2 2 2
 
*/
 
signed main(){
	ios_base::sync_with_stdio(0); cin.tie(0);
	int t = 1;
	//~ cin>>t;
	while(t--) solve();
}

컴파일 시 표준 에러 (stderr) 메시지

exam.cpp: In function 'void solve1(int, std::vector<int>&, std::vector<int>&)':
exam.cpp:41:7: warning: 'res' may be used uninitialized in this function [-Wmaybe-uninitialized]
   41 |   int res;
      |       ^~~

• Subtask 1: 14 / 14
• Subtask 2: 12 / 12
• Subtask 3: 13 / 13
• Subtask 4: 0 / 23
• Subtask 5: 16 / 16
• Subtask 6: 22 / 22