Submission #560969

#	Time	Username	Problem	Language	Result	Execution time	Memory
560969		Aktan	JJOOII 2 (JOI20_ho_t2)	C++14	100 / 100	12 ms	3300 KiB

ho_t2

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
using namespace std;
template<class T>
using ordered_set = tree<T, null_type,less<T>, rb_tree_tag,tree_order_statistics_node_update>;
#define int long long
#define ft first
#define sc second
const int mod=1e9+7,INF=1e17,N=1e5;

main(){
ios_base::sync_with_stdio(false);cin.tie(0);cout.tie(0);
 int n,k;
 cin >> n >> k;
 string s;
 cin >> s;
 int ans=INF;
 vector<int> a,b,c;
 for(int i=0;i<n;i++){
 	if(s[i]=='J'){
 		a.push_back(i);
	 }
	 else if(s[i]=='O'){
	 	b.push_back(i);
	 }
	 else{
	 	c.push_back(i);
	 }
 }
 for(int i=k-1;i<b.size();i++){
 	int l=b[i-k+1],r=b[i];
 	int x=--lower_bound(a.begin(),a.end(),l)-a.begin(),y=lower_bound(c.begin(),c.end(),r)-c.begin();	
 	x-=k;
 	x++;
 	y+=k;
 	y--;
 	if(x>=0 && y<c.size()){
 		ans=min(ans,(c[y]-a[x]+1)-(k*3));
	 }
 }
 if(ans==INF){
 	cout << -1;
 	return 0;
 }
 cout << ans;
}

Compilation message (stderr)

ho_t2.cpp:13:1: warning: ISO C++ forbids declaration of 'main' with no type [-Wreturn-type]
   13 | main(){
      | ^~~~
ho_t2.cpp: In function 'int main()':
ho_t2.cpp:32:17: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   32 |  for(int i=k-1;i<b.size();i++){
      |                ~^~~~~~~~~
ho_t2.cpp:39:15: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<long long int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   39 |   if(x>=0 && y<c.size()){
      |              ~^~~~~~~~~

• Subtask 1: 1 / 1
• Subtask 2: 12 / 12
• Subtask 3: 87 / 87