Структуры, heap, C++ Опции

[first_day]

Передо мной стоит следующая задача:

Есть (N<10^5) чисел, на каждом шаге алгоритма нужно находить 2 минимальных числа и заменять их на их сумму. Так делается до тех пор, пока не останется одно число. Сделать то я сделал, вот только работает это долго.

Знаю, что есть такая структура данных как куча, которая выдает минимальный элемент за O(1) и добавляет за O(log N). Но как это реализовывать не знаю, не нашел. А то, что находил, было совсем не понятно... возможно потому, что никогда не работал со структурами.

Может это можно реализовать на простом массиве?

Помогите, кто может.

P.S.
Вот как я делал:

#include <fstream>
#include <vector>
#include <algorithm>
#include <iomanip>
using namespace std;
int main()
{
	ifstream in("input.txt");
	ofstream out("output.txt");
	out.setf(ios::fixed);
	out.precision(2);
	double sum=0,z;
	int n,i,j,d=0;
	vector <int> v;
	in>>n;
	for(i=0;i<n;i++)
	{
		in>>j;
		v.push_back(j);
		d++;
	}
	while (d>1)
	{
		sort(v.begin(),v.end());
		z=v[0]+v[1];
		sum+=z/100*5; //5% процентов от полученной суммы двух этих минимальных чисел
		v.push_back(z);
		v.erase(v.begin(),v.begin()+2);
		d--;
	}
	out<<sum;
	return 0;
}

Сообщение отредактировано: first_day - 2.05.2008 20:54

[volvo]

Непонятна лишь одна строчка:

Смотри... В файле queue приоритетная очередь определяется вот так:

   namespace std {
       template <class T,
                 class Container = vector<T>,
                 class Compare = less<typename Container::value_type> >
       class priority_queue;
   }

, то есть, первый параметр - это тип элементов, с которым будет работать priority_queue, второй - тип контейнера, в котором все данные будут храниться внутри priority_queue (по умолчанию - vector<того_же_типа>), а третий - критерий сортировки. По умолчанию используется less<тип_элемента_контейнера>, тебе понадобилось "обратное" поведение, значит, меняем на greater<double>.

После этого все функции действуют наоборот(т.е. min вместо max)?

Не все... Это будет заметно только при работе с priority_queue, и будет выражаться в том, что выдаваться результаты будут не по убыванию (сначала - бОльшие, потом - меньшие), а по возрастанию значений...

Сообщение отредактировано: volvo - 2.05.2008 23:39