优先队列 priority_queue 用法 (struct,非cmp)

#include<iostream>
#include<functional>
#include<queue>
#include<vector>
using namespace std;


struct cmp1
{
	bool operator () (int &a, int &b)
	{
		return a > b ;              // 从小到大排序,值 小的 优先级别高
	}
};

struct cmp2
{
	bool operator () (int &a, int &b)
	{
		return a < b;             // 从大到小   int默认为降序
	}
};

struct number1
{
	int x;
	bool operator < (const number1 &a)const 
	{
		return x > a.x;         // 从小到大  ,x 小的 优先级别高
	}
};

struct number2
{
	int x;
	bool operator < (const number2 &a)const
	{
		return x < a.x;            // 从大到小  ,x 大的优先级别高
	}
};

int a[] = {14,10,56,7,83,22,36,91,3,47,72,0};
number1 num1[] ={14,10,56,7,83,22,36,91,3,47,72,0};
number2 num2[] ={14,10,56,7,83,22,36,91,3,47,72,0};

int main()
{
	priority_queue<int>que;  // 采用默认优先级构造队列  从大到小。 

        priority_queue<int, vector<int>, cmp1 >que1;
	priority_queue<int, vector<int>, cmp2 >que2;

	priority_queue<int, vector<int>, greater<int> > que3;  //functional 头文件自带的 
	priority_queue<int, vector<int>, less<int> > que4;      //functional 头文件自带的 

	priority_queue<number1> que5;
	priority_queue<number2> que6;


	int i;
	for(i=0;a[i];i++)
	{
		que.push(a[i]);
		que1.push(a[i]);
		que2.push(a[i]);
		que3.push(a[i]);
		que4.push(a[i]);

	}

	 for(i=0;num1[i].x;i++)  
             que5.push(num1[i]);  
        for(i=0;num2[i].x;i++)  
             que6.push(num2[i]);  

	printf("采用默认优先关系:\n(priority_queue<int>que;)\n");
	printf("Queue 0:\n");
        while(!que.empty())
	{
		printf("%3d",que.top());
		que.pop();
	}
	puts("");
	puts("");

    printf("采用结构体自定义优先级方式一:\n(priority_queue<int,vector<int>,cmp>que;)\n");
	printf("Queue 1:\n");
       while(!que1.empty())
	{
		printf("%3d",que1.top());
		que1.pop();
	}
	puts("");

	printf("Queue 2:\n");
        while(!que2.empty())
	{
		printf("%3d",que2.top());
		que2.pop();
	}
	puts("");
	puts("");

	printf("采用头文件\"functional\"内定义优先级:\n(priority_queue<int, vector<int>,greater<int>/less<int> >que;)\n");
	printf("Queue 3:\n");
	while(!que3.empty())
	{
		printf("%3d",que3.top());
		que3.pop();
	}
    puts("");

	printf("Queue 4 :\n");
	while(!que4.empty())
	{
		printf("%3d",que4.top());
		que4.pop();
	}
	puts("");
	puts("");

	printf("采用结构体自定义优先级方式二:\n(priority_queue<number>que)\n");
	printf("Queue 5:\n");
	while(!que5.empty())
	{
		printf("%3d",que5.top());
		que5.pop();
	}
	puts("");

	printf("Queue 6:\n");
	while(!que6.empty())
	{
		printf("%3d",que6.top());
		que6.pop();
	}

	system("pause");
	return 0;
}

Problem Description

有1<=n<=20000个木棒,要把这n个木棒还原为一根并且每次只可以两个相连接,费用就是每次要接的两个木棒的长度和,要求就是把这个棒还原,要求费用最小。

Input

输入包含多组测试数据,每组测试数据第一行是一个正整数n,接下来n行,每行一个数字,代表木棒长度。

Output

对于每组测试数据,输出一行正整数,表示把棒还原的最小费用。

Sample Input

3
8
5
8

Sample Output

34

#include <iostream>
#include <stdio.h>
#include <queue>
using namespace std; 


struct cmp{
    bool operator()(int &a, int &b)
    {
        return a>b;  
    }    
};


int total;
priority_queue<int,vector<int>,cmp>my;


int solve(){
    int a, b, sum=0;
    while(1){
        a=my.top();
        my.pop();
        b=my.top();
        my.pop();
        sum+=(a+b);
        if(a+b==total)
            break;
        my.push(a+b);
    }    
    return sum;
}




int 
main()
{
    int k, temp;
    while(cin>>k){
        total=0;
        while(my.empty()!=1)
            my.pop();
        while(k--){
            cin>>temp;
            total+=temp;
            my.push(temp);
        }
        int ans;
        ans=solve();
        cout<<ans<<endl;
    }    
    return 0;
}

 
 
 
 
 
 
  
  
  
  

你可能感兴趣的:(优先队列 priority_queue 用法 (struct,非cmp))