Submission #892803

# Submission time Handle Problem Language Result Execution time Memory
892803 2023-12-26T02:22:00 Z ReLice Chorus (JOI23_chorus) C++14
16 / 100
7000 ms 600 KB
#include <bits/stdc++.h>
#define ll long long
#define str string
#define ins insert
#define ld long double
#define pb push_back
#define pf push_front
#define pof pop_front()
#define pob pop_back()
#define lb lower_bound
#define ub upper_bound
#define endl "\n"
#define fr first
#define sc second
#define all(x) x.begin(),x.end()
#define rall(x) x.rbegin(),x.rend()
#define sz size()
#define bc back()
#define ar array
#define vll vector<ll> 
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
template <class _T>
bool chmin(_T &x, const _T &y){
    if(x>y){
        x=y;
        return true;
    }
    return false;
}
template <class _T>
bool chmax(_T &x, const _T &y){
    bool flag=false;
    if (x<y){
        x=y;flag|=true;
    }
    return flag;
}
#define ordered_set tree<ll, null_type,less_equal<ll>, rb_tree_tag,tree_order_statistics_node_update>
void fre(string s){freopen((s+".in").c_str(),"r",stdin);freopen((s+".out").c_str(),"w",stdout);}
void start(){
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
}
const ll inf=2e18+7;
const ll mod=1e9+7;
const ll N=1e5+7;
const ld eps=1e-9;
void solve(){
	ll i,j;
	ll n,k;
	str s;
	cin>>n>>k;
	cin>>s;
	str os=s;
	ll ans=inf;
	set<ll> as,bs;
	for(ll mask=0;mask<(1ll<<s.sz);mask++){
		ll c=0,c2=0;
		str ss;
		for(i=0;i<n*2;i++){
			if(mask&(1ll<<i)){
				c++;
				ss+='A';
			}
			else {
				ss+='B';
				c2++;
			}
		}
		if(c!=n)continue;
		c=c2=0;
		j=0;
		for(i=0;i<k;i++){
			while(j<n*2 && ss[j]=='A'){
				c++;
				j++;
			}
			while(j<n*2 && c){
				if(ss[j]=='A'){
					c2++;
				}else c--;
				j++;
			}
			c=c2;
			c2=0;
		}
		if(j<n*2)continue;
		for(i=0;i<n*2;i++){
			if(s[i]=='A')as.ins(i);
			else bs.ins(i);
		}
		ll sum=0;
		for(i=0;i<n*2;i++){
			if(s[i]!=ss[i]){
				if(s[i]=='A'){
					ll x=*bs.begin();
					sum+=(x-i);
					swap(s[i],s[x]);
					as.ins(x);
					as.erase(as.begin());
				}else{
					ll x=*as.begin();
					sum+=(x-i);
					swap(s[i],s[x]);
					bs.ins(x);
					bs.erase(bs.begin());
				}
			}
			if(s[i]=='A')as.erase(as.begin());
			else bs.erase(bs.begin());
		}
		ans=min(ans,sum);
		s=os;
	}
	cout<<ans<<endl;
}
signed main(){
	start();
    ll t=1;
	//cin>>t;
    while(t--) solve();
    return 0;
}
/*
1
6
5 4 3 2 1 0





*/

Compilation message

chorus.cpp: In function 'void fre(std::string)':
chorus.cpp:42:27: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   42 | void fre(string s){freopen((s+".in").c_str(),"r",stdin);freopen((s+".out").c_str(),"w",stdout);}
      |                    ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~
chorus.cpp:42:64: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   42 | void fre(string s){freopen((s+".in").c_str(),"r",stdin);freopen((s+".out").c_str(),"w",stdout);}
      |                                                         ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 136 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 2 ms 344 KB Output is correct
7 Correct 117 ms 428 KB Output is correct
8 Correct 114 ms 596 KB Output is correct
9 Correct 107 ms 432 KB Output is correct
10 Correct 108 ms 436 KB Output is correct
11 Correct 145 ms 348 KB Output is correct
12 Correct 137 ms 432 KB Output is correct
13 Correct 143 ms 428 KB Output is correct
14 Correct 110 ms 600 KB Output is correct
15 Correct 150 ms 348 KB Output is correct
16 Correct 151 ms 432 KB Output is correct
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 136 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 2 ms 344 KB Output is correct
7 Correct 117 ms 428 KB Output is correct
8 Correct 114 ms 596 KB Output is correct
9 Correct 107 ms 432 KB Output is correct
10 Correct 108 ms 436 KB Output is correct
11 Correct 145 ms 348 KB Output is correct
12 Correct 137 ms 432 KB Output is correct
13 Correct 143 ms 428 KB Output is correct
14 Correct 110 ms 600 KB Output is correct
15 Correct 150 ms 348 KB Output is correct
16 Correct 151 ms 432 KB Output is correct
17 Execution timed out 7027 ms 348 KB Time limit exceeded
18 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 136 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 2 ms 344 KB Output is correct
7 Correct 117 ms 428 KB Output is correct
8 Correct 114 ms 596 KB Output is correct
9 Correct 107 ms 432 KB Output is correct
10 Correct 108 ms 436 KB Output is correct
11 Correct 145 ms 348 KB Output is correct
12 Correct 137 ms 432 KB Output is correct
13 Correct 143 ms 428 KB Output is correct
14 Correct 110 ms 600 KB Output is correct
15 Correct 150 ms 348 KB Output is correct
16 Correct 151 ms 432 KB Output is correct
17 Execution timed out 7027 ms 348 KB Time limit exceeded
18 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 136 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 2 ms 344 KB Output is correct
7 Correct 117 ms 428 KB Output is correct
8 Correct 114 ms 596 KB Output is correct
9 Correct 107 ms 432 KB Output is correct
10 Correct 108 ms 436 KB Output is correct
11 Correct 145 ms 348 KB Output is correct
12 Correct 137 ms 432 KB Output is correct
13 Correct 143 ms 428 KB Output is correct
14 Correct 110 ms 600 KB Output is correct
15 Correct 150 ms 348 KB Output is correct
16 Correct 151 ms 432 KB Output is correct
17 Execution timed out 7027 ms 348 KB Time limit exceeded
18 Halted 0 ms 0 KB -
# Verdict Execution time Memory Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 348 KB Output is correct
4 Correct 136 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 2 ms 344 KB Output is correct
7 Correct 117 ms 428 KB Output is correct
8 Correct 114 ms 596 KB Output is correct
9 Correct 107 ms 432 KB Output is correct
10 Correct 108 ms 436 KB Output is correct
11 Correct 145 ms 348 KB Output is correct
12 Correct 137 ms 432 KB Output is correct
13 Correct 143 ms 428 KB Output is correct
14 Correct 110 ms 600 KB Output is correct
15 Correct 150 ms 348 KB Output is correct
16 Correct 151 ms 432 KB Output is correct
17 Execution timed out 7027 ms 348 KB Time limit exceeded
18 Halted 0 ms 0 KB -