洛谷 P6033 [NOIP2004 提高组] 合并果子 加强版

[NOIP2004 提高组] 合并果子 加强版

题目背景

本题除【数据范围与约定】外与 P1090 完 全 一 致。

题目描述

在一个果园里，多多已经将所有的果子打了下来，而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。

每一次合并，多多可以把两堆果子合并到一起，消耗的体力等于两堆果子的重量之和。可以看出，所有的果子经过 $(n-1)$ 次合并之后， 就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。

因为还要花大力气把这些果子搬回家，所以多多在合并果子时要尽可能地节省体力。假定每个果子重量都为 $1$，并且已知果子的种类数和每种果子的数目，你的任务是设计出合并的次序方案，使多多耗费的体力最少，并输出这个最小的体力耗费值。

例如有 $3$ 堆果子，数目依次为 $1,2,9$。可以先将 $1$、$2$ 堆合并，新堆数目为 $3$，耗费体力为 $3$。接着，将新堆与原先的第三堆合并，又得到新的堆，数目为 $12$，耗费体力为 $12$。所以多多总共耗费体力为 $3+12=15$。可以证明 $15$ 为最小的体力耗费值。

输入格式

输入的第一行是一个整数 $n$，代表果子的堆数。
输入的第二行有 $n$ 个用空格隔开的整数，第 $i$ 个整数代表第 $i$ 堆果子的个数 $a_i$。

输出格式

输出一行一个整数，表示最小耗费的体力值。

样例 #1

样例输入 #1

3 
1 2 9

样例输出 #1

15

提示

【数据规模与约定】

本题采用多测试点捆绑测试，共有四个子任务。

• Subtask 1（10 points）：$1\le n\le 8$。
• Subtask 2（20 points）：$1\le n\le 10^3$。
• Subtask 3（30 points）：$1\le n\le 10^5$。
• Subtask 4（40 points）：$1\le n\le 10^7$。

对于全部的测试点，保证 $1\le a_i\le 10^5$。

【提示】

• 请注意常数因子对程序效率造成的影响。
• 请使用类型合适的变量来存储本题的结果。
• 本题输入规模较大，请注意数据读入对程序效率造成的影响。

分析：

和他弟P1090 [NOIP2004 提高组] 合并果子一样，我们也是用贪心的思想来做。

贪心的思路应该都了解了吧，就是每一次选取最小的两堆果子进行合并。

怎么去取呢？
我们可以用优先队列啊····但是这里要求是O(n)的算法，要超时。

鱼逝：
我们可以先排序，选取最小的两堆果子，然后合并，插入。可是插入的效率实在是太低了，我们还需要优化。

又鱼逝：
我们可以把需要插入的点用一个队列 q 存储起来。但由于需要插入的点肯定会越来越大相当于延迟插入。所以在我们目标插入点就是我们当前最小的那一堆的时候，我们就可以把他插入进来。

种杰：
我们先建立两个队列q，qq，再用桶排序。排序结果放进q队列当中，合并结果放在qq队列当中，每次选从两个队列的最前面选取比较小的合并。

代码（QWQ）：

#include<bits/stdc++.h>

using namespace std;

queue <int> q,qq;

int a[1000000];

int n;

int x,y;

int main() 
{
	cin>>n;
	
	for(int i=1;i<=n;i++) 
	{
		cin>>x;
		
		a[x]++;
	}
	
	for(int i=1;i<=100000;i++) 
	{
		while(a[i]>0) 
		{
			a[i]--;
			
			q.push(i);
		}
	}
	
	int sum=0;
	
	for(int i=1;i<n;i++) 
	{
		if((q.front()<qq.front()&&q.empty()==0)||qq.empty()==1) 
		{
			x=q.front();
			
			q.pop();
		}
		else 
		{
			x=qq.front();
			
			qq.pop();
		}
		
		if((q.front()<qq.front()&&q.empty()==0)||qq.empty()==1) 
		{
			y=q.front();
			
			q.pop();
		}
		else 
		{
			y=qq.front();
			
			qq.pop();
		}
		
		sum=sum+(x+y);
		
		int s=x+y;
		
		qq.push(s);
	} 
	
	cout<<sum;
	
	return 0;
} 

很简单是吧，30分······

要开longlong！！！

代码（QAQ）：

#include<bits/stdc++.h>

#define int long long

using namespace std;

queue <int> q,qq;

int a[1000000];

int n;

int x,y;

signed main() 
{
	cin>>n;
	
	for(int i=1;i<=n;i++) 
	{
		cin>>x;
		
		a[x]++;
	}
	
	for(int i=1;i<=100000;i++) 
	{
		while(a[i]>0) 
		{
			a[i]--;
			
			q.push(i);
		}
	}
	
	int sum=0;
	
	for(int i=1;i<n;i++) 
	{
		if((q.front()<qq.front()&&q.empty()==0)||qq.empty()==1) 
		{
			x=q.front();
			
			q.pop();
		}
		else 
		{
			x=qq.front();
			
			qq.pop();
		}
		
		if((q.front()<qq.front()&&q.empty()==0)||qq.empty()==1) 
		{
			y=q.front();
			
			q.pop();
		}
		else 
		{
			y=qq.front();
			
			qq.pop();
		}
		
		sum=sum+(x+y);
		
		int s=x+y;
		
		qq.push(s);
	} 
	
	cout<<sum;
	
	return 0;
} 

额······TLE了，60分

这种时候一定要用快读！！！

快速读入：

int read()
{
	int res=0;
	
	char ch=getchar();
	
	while(ch<'0'||ch>'9') 
	{
		ch=getchar();
	}
	
	while(ch<='9'&&ch>='0') 
	{
		res=res*10+ch-'0';
		
		ch=getchar();
	}
	
	return res;
}

代码 (＾－＾)V：

#include<bits/stdc++.h>

#define int long long

using namespace std;

queue <int> q,qq;

int a[1000000];

int n;

int x,y;

int read()
{
	int res=0;
	
	char ch=getchar();
	
	while(ch<'0'||ch>'9') 
	{
		ch=getchar();
	}
	
	while(ch<='9'&&ch>='0') 
	{
		res=res*10+ch-'0';
		
		ch=getchar();
	}
	
	return res;
}

signed main() 
{
	n=read();
	
	for(int i=1;i<=n;i++) 
	{
		x=read();
		
		a[x]++;
	}
	
	for(int i=1;i<=100000;i++) 
	{
		while(a[i]>0) 
		{
			a[i]--;
			
			q.push(i);
		}
	}
	
	int sum=0;
	
	for(int i=1;i<n;i++) 
	{
		if((q.front()<qq.front()&&q.empty()==0)||qq.empty()==1) 
		{
			x=q.front();
			
			q.pop();
		}
		else 
		{
			x=qq.front();
			
			qq.pop();
		}
		
		if((q.front()<qq.front()&&q.empty()==0)||qq.empty()==1) 
		{
			y=q.front();
			
			q.pop();
		}
		else 
		{
			y=qq.front();
			
			qq.pop();
		}
		
		sum=sum+(x+y);
		
		int s=x+y;
		
		qq.push(s);
	} 
	
	cout<<sum;
	
	return 0;
} 

结束啦~~~