Submission #595665

#	Time	Username	Problem	Language	Result	Execution time	Memory
595665		inksamurai	Event Hopping 2 (JOI21_event2)	C++17	0 / 100	19 ms	2772 KiB

event2

#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) for(int i=0;i<n;i++)
#define rng(i,c,n) for(int i=c;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define fi first
#define se second
#define pb push_back
#define sz(a) (int)a.size()
#define vec(...) vector<__VA_ARGS__>
#define _3uHasSr ios::sync_with_stdio(0),cin.tie(0)
typedef long long ll;
using pii=pair<int,int>;
using vi=vector<int>;
void print(){cout<<'\n';}
template<class h,class...t>
void print(const h&v,const t&...u){cout<<v<<' ',print(u...);}
// e

using T=pair<pii,int>;

signed main(){
_3uHasSr;
	int n,k;
	cin>>n>>k;
	vec(T) a(n);
	rep(i,n){
		cin>>a[i].fi.fi>>a[i].fi.se;
		a[i].se=i;
	}
	assert(n<=3000);
	vec(pii) _a;
	rep(i,n){
		_a.pb(a[i].fi);
	}
	sort(a.begin(), a.end(),[&](const T&l,const T&r){
		return l.fi.se<r.fi.se;
	});
	vec(pii) stk;
	vi pns;
	rep(_i,n){
		pii _p=_a[_i];
		vec(pii) _stk=stk;
		stk.pb(_p);
		sort(stk.begin(), stk.end(),[&](pii l,pii r){
			return l.se<r.se;
		});
		bool pok=1;
		// check if i already intersect 
		{
			int _lst=-1;
			for(auto p:stk){
				if(p.fi<_lst){
					pok=0;
					break;
				}
				_lst=p.se;
			}
		}
		if(!pok){
			stk=_stk;
			continue;
		}
		int _lst=-1;
		int j=0,cnt=0;
		rep(i,n){
			pii p=a[i].fi;
			// if(_i==1){
			// 	print(p.fi,p.se,j);
			// }
			if(p==stk[j]){
				_lst=stk[j].se;
				j+=1;
				cnt+=1;
			}else if(p.fi>=_lst and (j==sz(stk) or p.se<=stk[j].fi)){
				_lst=p.se;
				cnt+=1;
			}
		}
		// return 0;
		// print(_i,cnt);
		// if(_i==1){
		// 	return 0;
		// }
		// //
		stk=_stk;
		if(cnt>=k){
			pns.pb(_i);
			stk.pb(_p);
		}
	}
	while(sz(pns)>k){
		pns.pop_back();
	}
	if(sz(pns)<k){
		print(-1);
		return 0;
	}
	rep(i,k){
		print(pns[i]+1);
	}
// 
	return 0;
}

• Subtask 1: 0 / 7
• Subtask 2: 0 / 1
• Subtask 3: 0 / 31
• Subtask 4: 0 / 61