UVALive - 6434 (贪心)

#include<cstdio>
#include<algorithm>
#include<cstring>
#define INF 1e9+5
using namespace std;
int s[110];
int vt[110];//记录当前位置的数字是属于哪个集合的 
int main(){
	int t;
	int n,m;
	int i,j,k;
	int a,b,c;
	int mn;
	int nb=0;
	scanf("%d",&t);
	while(t--){
		nb++;
		scanf("%d%d",&n,&m);
		for(i=0;i<n;i++){
			scanf("%d",&s[i]);
			vt[i]=i;//分成n组 
		}
		if(m==n)
		printf("Case #%d: 0\n",nb);
		else{
			sort(s,s+n);
			for(i=n;i>m;i--){
				mn=INF;
				a=vt[0];
				for(j=1;j<n;j++){
					if(vt[j]!=a){//两个分组的交界位置 
						if(s[j]-s[j-1]<=mn){//找把两个集合合并在一起增加消耗最小的位置 ，一开始的思路只是找出合并后最大值和最小值之差最小的两个组进行合并 
						//结果wa了一发，仔细想想后发现这样合并后并不能把增加的消耗值降到最小
						/*
						1
						10 2
						1 5 6 7 8 9 10 13 13 13
						当分组数位3的时候
						此时分组为 （1） （5，6，7，8，9，10） （13，13，13）
						如果只是单纯的把合并后最大值和最小值之差最小的两组合并的话，
						 （1，5，6，7，8，9，10）   （13，13，13）
						 此时总消耗为9
						 （1） （5，6，7，8，9，10，13，13，13）
						 而最小的总消耗却是8 
						不能保证每次的增长量最小 
						*/ 
							c=j;//记录分界点 
							mn=s[j]-s[j-1];
						}
						a=vt[j];//连接相邻的两个组的消耗一定是最小的 
					}
				}	
				/*for(j=0;j<n;j++){//不太理解的同学可以输出一下每次的分组情况便于理解 
					printf("%d ",vt[j]);
				}
				printf("\n");*/
				int y=vt[c];
				for(j=c;vt[j]==y;j++){//合并 
					vt[j]=vt[c-1];
				}
			}
			a=vt[0];
			b=0;
			long long sum=0;
			for(i=1;i<n;i++){
				if(vt[i]!=a){
					sum+=s[i-1]-s[b];//计算消耗值 
					b=i;
					a=vt[i];
				}
			}
			sum+=s[n-1]-s[b];
			printf("Case #%d: %lld\n",nb,sum);
		}
		
	}
	return 0;
}

#include<stdio>
#include<algorithm>
using namespace std;
int s[110]
int main(){
	int t,n,m;
	int i,j,nb;
	nb=0;
	int a,b;
	scanf("%d",&t);
	while(t--){
		nb++;
		scanf("%d%d",&n,&m);
		for(i=0;i<n;i++){
			scanf("%d",&s[i]);	
		}
		sort(s,s+n);
		for(i=0;i<n-1;i++)//直接记录下所有相邻的数字要合并在一起的消耗（这和上个代码的核心思想一致，但是
		//少了许多不必要的操作，直接记录下最核心的数据） 
		s[i]=s[i+1]-s[i];
		sort(s,s+n-1);
		int sum=0;
		for(i=0;i<n-m;i++){//每次找到消耗代价最小的那一次合并 
			sum+=s[i];
		}
		printf("Case #%d: %d",nb,sum);
	}
	return 0;
}