D1. Equalizing by Division (easy version)

原题
这道题因为$a_i,k,n$都比较小，对数据进行一些预处理就可以解决。
$$\begin{gathered} 1\le a_i\le 2\times 10^5 \\ 1\le k\le n \end{gathered}$$

解题步骤

1.定义矩阵$chu[i][j]$,表示$a_i$,除以2，连除$j$次后的结果 $\mathcal{O}(n)$
2.定义数组$cishu[p]$,记录$p=chu[i][j],cishu[p]$表示$p$在$chu$矩阵中出现的次数。$\mathcal{O}(n)$
3.对于$cishu[p]\ge k$,得到p需要的最少操作次数，在所有值中再取最小值。$\mathcal{O}(n^2)$
这道题n并不大，所以时间上不会超限，分析一下空间上也不会超限

代码实现

#include<bits/stdc++.h>
using namespace std;
#define MAX 55
#define MAXAI 200004
#define MI 24
int a[MAX];
int chu[MAX][MI]; //除 
int cishu[MAXAI];

int main(){
	int n,k;	
	scanf("%d%d",&n,&k);	
	for(int i=0;i<n;i++){
  		scanf("%d",&a[i]);
 	}
 	for(int i=0;i<n;i++){
 		int j=0;
  		chu[i][0]=a[i]; 
  		cishu[a[i]]++;
  		while(chu[i][j]!=0){
   			int tmp=chu[i][j]/2;
   			chu[i][j+1]=tmp;
   			cishu[tmp]++; 
   			j++;
  		}
 	} 
 	int Min=INT_MAX;
 	for(int i=0;i<MAXAI;i++){
 		if(cishu[i]>=k){
  	 		int ci=0;
   			int num=0;
   			for(int p=0;p<MI;p++){
    				for(int j=0;j<n;j++){
     					if(chu[j][p]==i){
      
      						if(p==0||p!=0&&chu[j][p-1]!=0)
       							ci++;  //避免重复计算chu[i][j]矩阵之后未使用的部分都是0
      						num+=p;
      						if(ci==k){
       							if(Min>num){
        							Min=num;
       							}
       							break;
      						}
    	 				}
    				}
   			}
  		}
 	}
 	printf("%d\n",Min);
 	return 0; 
 }