牛客oj演習10.2 K番目の小数を検索（セット）&&演習10.3フルーツを移動（ハフマンツリー）

 

もともとセットでやるのは面倒だと思っていたのですが、その後、一番シンプルな時計を考えるのはさっぱりせず、少しバランスが取れていました。。

 

 

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <set>
#include <queue>
#include <cmath>
#include <climits>
 
using namespace std;
 
const int MAXN = 1005;
const int INF = INT_MAX;

int main(){
//	freopen("in.txt", "r", stdin);
	int n, k, x;
	while(~scanf("%d", &n)){
		set<int> myset;
		for(int i = 0; i < n; i++){
			scanf("%d", &x);
			myset.insert(x);
		}
		scanf("%d", &k);
		set<int>::iterator it;
		int time = 1;
		for(it = myset.begin(); it != myset.end(); it++){
			if(time == k){
				cout << *it << endl;
				break;
			}
			time ++;
		}
	}
	return 0;
}

 

 

ハフマンツリーの水の問題。ところで優先度キューを確認しますが、頂点の値ではなく最大の重みを出力することを忘れないでください。

 

 

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <cmath>
#include <climits>
 
using namespace std;
 
const int MAXN = 1005;
const int INF = INT_MAX;

int main(){
//	freopen("in.txt", "r", stdin);
	int n, x;
	while(~scanf("%d", &n)){
		if(n == 0) break;
		priority_queue<int, vector<int>, greater<int> > myqueue;
		for(int i = 0; i < n; i++){
			scanf("%d", &x);
			myqueue.push(x);
		}
		int ans = 0;
		while(myqueue.size() > 1){
			int x1 = myqueue.top();
			myqueue.pop();
			int x2 = myqueue.top();
			myqueue.pop();
			ans += (x1 + x2);
			myqueue.push(x1 + x2);
		}
		printf("%d\n", ans);
	}
	return 0;
}