제출 #960129

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
960129		MilosMilutinovic	Cheerleaders (info1cup20_cheerleaders)	C++14	33 / 100	2013 ms	21596 KiB

cheerleaders

#include<bits/stdc++.h>
 
#define pb push_back
#define fi first
#define se second
#define mp make_pair
 
using namespace std;
 
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef long double ld;
 
template <typename T> bool chkmin(T &x,T y){return x>y?x=y,1:0;}
template <typename T> bool chkmax(T &x,T y){return x<y?x=y,1:0;}
 
ll readint(){
	ll x=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
 
int n;
int h[1<<20],cnt[1<<20],a[1<<20],b[1<<20],tmp[1<<20],ttmp[1<<20];
ll cst[20];
string str;
 
int ask(int p){
	int ret=0;
	while(p>=0){
		ret+=cnt[p];
		p=(p&(p+1))-1;
	}
	return ret;
}
 
void change(int p,int v){
	while(p<(1<<n)){
		cnt[p]+=v;
		p|=(p+1);
	}
}
 
ll solve(){
	for(int i=0;i<(1<<n);i++) cnt[i]=0;
	ll ret=0;
	for(int i=(1<<n)-1;i>=0;i--){
		ret+=ask(h[i]);
		change(h[i],1);
	}
	for(int i=0;i<(1<<n);i++) change(h[i],-1);
	return ret;
}

int cc[1<<20];

void op1(){
	for(int i=0;i<(1<<n);i++) cc[i^(1<<(n-1))]=h[i];
	for(int i=0;i<(1<<n);i++) h[i]=cc[i];
}
 
void op2(){
	int x=0,y=0;
	for(int i=0;i<(1<<n);i++){
		if(i%2==0) a[x]=h[i],x++;
		if(i%2==1) b[y]=h[i],y++;
	}
	int p=0,q=0;
	for(int i=0;i<(1<<n);i++){
		if(p<x) h[i]=a[p],p++;
		else h[i]=b[q],q++;
	}
}

void go(vector<int> v,int d){
	if(d<0) return;
	for(int i=v.size()-1;i>=0;i--){
		cst[d]-=ask(h[v[i]]);
		change(h[v[i]],1);
	}
	for(int i:v) change(h[i],-1);
	for(int i=v.size()-1;i>=0;i--){
		cst[d]+=ask(h[v[i]]^(1<<d));
		change(h[v[i]]^(1<<d),1);
	}
	for(int i:v) change(h[i]^(1<<d),-1);
	vector<int> vec[2];
	for(int i:v) vec[h[i]>>d&1].pb(i);
	go(vec[0],d-1);
	go(vec[1],d-1);
}
 
ll work(int x){
	for(int i=0;i<(1<<n);i++) tmp[i]=h[i];
	for(int iter=0;iter<x;iter++) op2();
	int msk=0;
	ll ret=solve();
	str="";
	for(int mask=0;mask<(1<<n);mask++){
		for(int i=0;i<(1<<n);i++) ttmp[i]=h[i];
		for(int i=0;i<(1<<n);i++) a[i^mask]=h[i];
		for(int i=0;i<(1<<n);i++) h[i]=a[i];
		chkmin(ret,solve());
		for(int i=0;i<(1<<n);i++) h[i]=ttmp[i];
	}
	for(int i=0;i<(1<<n);i++) h[i]=tmp[i];
	return ret;
}
 
int main(){
	n=readint();
	for(int i=0;i<(1<<n);i++) h[i]=readint();
	ll res=work(0);
	int bst=0;
	for(int i=1;i<n;i++){
		ll c=work(i);
		if(c<res) res=c,bst=i;
	}
	assert(res==work(bst));
	printf("%lld\n",res);
	for(char c:str) printf("%c",c);
	return 0;
}

컴파일 시 표준 에러 (stderr) 메시지

cheerleaders.cpp: In function 'll work(int)':
cheerleaders.cpp:99:6: warning: unused variable 'msk' [-Wunused-variable]
   99 |  int msk=0;
      |      ^~~

• Subtask 1: 8 / 16
• Subtask 2: 5 / 10
• Subtask 3: 20 / 25
• Subtask 4: 0 / 21
• Subtask 5: 0 / 28