Help me find a bug for solving the problem Moamen and k-subarrays in Codeforces Round #737 (Div. 2)

Revision en3, by 018429, 2021-08-25 09:42:31

About the problem 1557B,what's the matter with following code , I can't figure out why it is always failed in case 58 of third test,I am almost crushed @^-^@

#include <bits/stdc++.h>
using namespace std;

int main() {
	int t;
	cin>>t;
	while(t--){
		int n,k;
		cin >> n >> k;
		vector<int> a(n),b(n);
		for(int i = 0;i < n;i++){
			cin >> a[i];
			b[i] = a[i]; 
		}
		sort(b.begin(),b.end());
		map<int,int> nxt;
		for(int i = 0;i < n - 1;i++)nxt[b[i]] = b[i+1];
		//nxt[b[n-1]] = b[n-1];
		int seg = 1;
		for(int i = 1;i < n;i++)if(a[i] != nxt[a[i-1]])seg++;
		if(seg <= k){
			cout<<"YES"<<endl;
		}else{
			cout<<"NO"<<endl;
		}
	}
  	return 0;
}
Update:

OK,I just rewrite the code,and make a mistake again,when i make nxt[b[n-1] = b[n],I even pass 10th test,then I realize that the last number's next value will be value zero in map when I tempt to access by [].For Example,in the case n = 4,k = 1,arrays like: 3 0 1 2,the code will think this is non-decreasing,Finally,the problem is solved,so happy^_^

Tags #help me, #debug

History

 
 
 
 
Revisions
 
 
  Rev. Lang. By When Δ Comment
en3 English 018429 2021-08-25 09:42:31 2 Tiny change: 'en pass 10 test,then' -> 'en pass 10th test,then'
en2 English 018429 2021-08-25 09:41:04 208
en1 English 018429 2021-08-25 09:33:17 1199 Initial revision (published)