Submission #1038478

#	Time	Username	Problem	Language	Result	Execution time	Memory
1038478		PM1	Watermelon (INOI20_watermelon)	C++17	100 / 100	94 ms	43236 KiB

Main

#include <bits/stdc++.h>
using namespace std;
#define fr first
#define sc second
#define ll long long
const int mxn=1e5+5,szz=(1<<20);
int n,a[mxn],b[mxn],k,L[mxn],R[mxn],sz[mxn],mx[mxn];
vector<int>g[mxn],par[mxn];
set<int>s;
void solve(int m){
	if(s.size()==0 && m<n+1){
		cout<<-1;
		exit(0);
	}
	if(m==n+1){
		k--;
		if(k==0){
			for(int i=1;i<=n;i++)
				cout<<a[i]<<" ";
			cout<<'\n';
			exit(0);	
		}
		return ;
	}
	vector<int>v;
	for(auto i:s){
		v.push_back(i);
		if(v.size()==10)
			break;
	}
	for(auto i:v){
		a[i]=m;
		s.erase(i);
		for(auto j:par[i]){
			sz[j]--;
			if(sz[j]==0)
				s.insert(j);
		}
	
		solve(m+1);

		s.insert(i);
		for(auto j:par[i]){
			sz[j]++;
			if(sz[j]==1)
				s.erase(j);
		}
	}

}
struct segment{
	int val[szz];
	void build(int id,int L,int R){
		if(L+1==R){
			val[id]=b[L];
			return ;
		}
		int mid=(L+R)>>1;
		build(id*2,L,mid);
		build(id*2+1,mid,R);
		val[id]=max(val[id*2],val[id*2+1]);
	}
	int get(int id,int L,int R,int l,int r){
		if(L>=r || l>=R)
			return 0;
		if(l<=L && R<=r)
			return val[id];
		int mid=(L+R)>>1;
		return max(get(id*2,L,mid,l,r),get(id*2+1,mid,R,l,r));
	}
}seg;
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	//freopen("gen.txt","r",stdin);
	//freopen("gh.txt","w",stdout);
	cin>>n>>k;
	int cnt=0;
	vector<int>hey;
	for(int i=1;i<=n;i++){
		cin>>b[i];
		if(b[i]==-1)hey.push_back(i);
	}
	if(b[n]!=-1){
		cout<<-1;
		return 0;
	}
	for(int i=1;i<hey.size();i++){
		par[hey[i]].push_back(hey[i-1]);
		g[hey[i-1]].push_back(hey[i]);
	}
	seg.build(1,1,n+1);
	for(int i=0;i<hey.size();i++){
		int l=(i==0)?1:hey[i-1]+1;
		vector<int>v;
		for(int j=hey[i]-1;j>=l;j--){
			par[j].push_back(hey[i]);
			g[hey[i]].push_back(j);
			R[j]=j+1;
			while(R[j]<hey[i] && b[R[j]]<b[j]){
				if(seg.get(1,1,n+1,j+1,R[j])<b[R[j]]){
					par[R[j]].push_back(j);
					g[j].push_back(R[j]);
				}
				
				R[j]=R[R[j]];
			}
			par[j].push_back(R[j]);
			g[R[j]].push_back(j);
			if(seg.get(1,1,n+1,j+1,R[j])+1!=b[j]){
				cout<<-1;
				return 0;
			}
		}
	}
	for(int i=1;i<=n;i++){
		sz[i]=g[i].size();
		if(sz[i]==0)
			s.insert(i);
	}
	solve(1);
	cout<<-1;
	return 0;
}

Compilation message (stderr)

Main.cpp: In function 'int main()':
Main.cpp:89:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   89 |  for(int i=1;i<hey.size();i++){
      |              ~^~~~~~~~~~~
Main.cpp:94:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   94 |  for(int i=0;i<hey.size();i++){
      |              ~^~~~~~~~~~~
Main.cpp:79:6: warning: unused variable 'cnt' [-Wunused-variable]
   79 |  int cnt=0;
      |      ^~~

• Subtask 1: 7 / 7
• Subtask 2: 25 / 25
• Subtask 3: 31 / 31
• Subtask 4: 18 / 18
• Subtask 5: 19 / 19