【codeforces】【速】931C

C. Laboratory Work

原題鏈接：931C. Laboratory Work
題意：給定一個最大值與最小值相差不超過2的數組，求一個平均值相同但數據相同個數最少的數組，並且需要輸出他相同數據的個數，允許重複值

題解：如代碼所示，利用桶，只有兩種情況

1. a–>2b<–c — 變成 — 2b
2. a<–2b–>c — 變成 — a c

考慮這兩種情況中的最小值就好了

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

int a[100001],mark[3];
int main(){
	int n;
	while(~scanf("%d",&n)){
		memset(mark,0,sizeof(mark));
		for(int i=0;i<n;i++)scanf("%d",&a[i]);
		sort(a,a+n);
		for(int i=0;i<n;i++)mark[a[i]-a[0]]++;
		int m_m = min(mark[0],mark[2]);
		if(mark[0]&&mark[1]&&mark[2]&&mark[1]/2>m_m){
			printf("%d\n",n-mark[1]/2*2);
			mark[0]+=mark[1]/2;
			mark[2]+=mark[1]/2;
			mark[1]%=2;
		}else{
			printf("%d\n",n-m_m*2);
			mark[0]-=m_m;
			mark[2]-=m_m;
			mark[1]+=2*m_m;
		}
		for(int i=0;i<3;i++)
			for(int j=0;j<mark[i];j++)
				printf("%d ",a[0]+i);
		putchar('\n');
	}
}