Cod sursa(job #2806069)

Utilizator AlexCrpCarpineanu Alexandru AlexCrp Data 22 noiembrie 2021 12:16:42
Problema Algoritmul lui Dijkstra Scor 20
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 17.36 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <stack>
#include <algorithm>
#include <set>
#include <limits>

using namespace std;

ifstream fin("dijkstra.in");
ofstream fout("dijkstra.out");

void countSort(vector<int>&);
vector<int> costuri;

class Graf
{
protected:

	int nrMuchii, nrNoduri;
	vector<vector<int > > listaAdiacenta;
	bool orientat;
private:
	void adaugaMuchie(int, int, bool);		    //Am decis sa fac adaugarea unei muchii in lista de adiacenta o functie, poate o sa fie useful later?
	void DFSAux(int, vector<bool>&);			//Functie auxiliara pt problema DFS de pe infoarena
	void BCXAux(int, int&, vector<int>&, vector<int>&, vector<int>&, stack<int>&, vector<vector<int>>&);		//Metoda auxiliara pt problema Componente Biconexe de pe infoarena
	void CTCAux(int, int&, vector<int>&, vector<int>&, vector<bool>&, stack<int>&, vector<vector<int>>&);		//Metoda auxiliara pt CTC
	void MuchiiAux(int, int&, vector<int>&, vector<int>&, vector<int>&, vector<bool>&, vector<vector<int>>&);	//Metoda auxiliara pt MuchiiCrit
	void DFSTopological(int, vector<bool>&, stack<int>&);			//Metoda auxiliara pt Sortarea topologica
	friend void countSort(vector<int>&);
public:
	Graf();
	Graf(int, int, bool);
	void setListaAdiacenta();
	void BFS(int);							//Metoda care rezolva problema BFS de pe infoarena
	int DFSrezolvare(int);					//Metoda care rezolva problema DFS de pe infoarena
	void Biconexe();						//Metoda care rezolva problema Componente Biconexe de pe infoarena
	void HavelHakimi();						//Metoda care pune in practica Havel Hakimi	
	void CTC();								//Metoda care afiseaza componentele tare conexe ale grafului	
	void MuchiiCrit();						//Metoda care afiseaza muchiile critice ale grafului
	void SortareTopologica();				//Metoda care sorteaza topologic nodurile grafului
};


class GrafPonderat:public Graf
{
protected:
	vector<pair<pair<int, int>, int>> reprezentareGrafPonderat;
	vector<vector<pair<int, int>>> listaAdiacentaPonderata;
	void KruskallReuniune(vector<int>&, int, int);
	int KruskallFindMultime(vector<int>&, int);
	static bool KruskallComp(int m1, int m2)
	{
		return costuri[m1] < costuri[m2];
	};


public:
	GrafPonderat();
	GrafPonderat(int, int, bool);
	void Kruskall();
	void PaduriMultimiDisjuncte();
	void Dijkstra(int);
	void setListaAdiacentaPonderata();
	void setReprezentareGrafPonderat();
};

void GrafPonderat::setListaAdiacentaPonderata()
{
	int x, y, z;
	listaAdiacentaPonderata.resize(nrMuchii + 1);
	for (int i = 0; i < nrMuchii; i++)
	{
		fin >> x >> y >> z;
		listaAdiacentaPonderata[x].push_back(make_pair(y, z));
	}
	
}

void GrafPonderat::setReprezentareGrafPonderat()
{
	int x, y, z;

	for (int i = 0; i < nrMuchii; i++)
	{
		fin >> x >> y >> z;
		reprezentareGrafPonderat.push_back(make_pair(make_pair(x, y), z));
	}
}

void GrafPonderat::Dijkstra(int nodSursa)
{
	set<pair<int, int>> heap;
	heap.insert({ 0, nodSursa });

	vector<int> distante(nrNoduri+1, 2000000);
	distante[nodSursa] = 0;

	while (!heap.empty())
	{
		pair<int, int> nodCurent = *heap.begin();
		heap.erase(heap.begin());

		for (auto nodAdiacent : listaAdiacentaPonderata[nodCurent.second])
		{
			if (distante[nodAdiacent.first] > distante[nodCurent.second] + nodAdiacent.second)
			{
				if (distante[nodAdiacent.first] != 2000000)
				{
					heap.erase(heap.find(make_pair(distante[nodAdiacent.first], nodAdiacent.first)));
				}
				distante[nodAdiacent.first] = distante[nodCurent.second] + nodAdiacent.second;
				heap.insert(make_pair(distante[nodAdiacent.first], nodAdiacent.first));
			}
		}
	}

	for (int i = 2; i <= nrNoduri; i++)
	{
		fout << distante[i] << ' ';
	}
}

void GrafPonderat::PaduriMultimiDisjuncte()
{
	cout << 'a';
	int n, m;
	fin >> n >> m;
	vector<int> multime(n+1, -1);
	vector<int> indici;
	cout << 'a';
	for (int i = 1; i < n+1; i++)
	{
		multime[i] = i;
	}
	cout << 'a';
	for (int i = 1; i <= m; i++)
	{
		
		int operatie;
		fin >> operatie;
		if (operatie == 1)
		{
			cout << 'b';
			int x, y;
			fin >> x >> y;
			KruskallReuniune(multime, x, y);
		}
		
		else
		{
			cout << 'c';
			int x, y;
			fin >> x >> y;

			if (KruskallFindMultime(multime, x) == KruskallFindMultime(multime, y))
			{
				fout << "DA\n";
			}
			else
			{
				fout << "NU\n";
			}
		}
	}
}

GrafPonderat::GrafPonderat() :Graf() {};

GrafPonderat::GrafPonderat(int nrN, int nrM, bool orr) :Graf(nrN, nrM, orr) {};

void GrafPonderat::KruskallReuniune(vector<int>& multime, int i, int j)
{
	multime[KruskallFindMultime(multime, i)] = KruskallFindMultime(multime, j);
}

int GrafPonderat::KruskallFindMultime(vector<int>& multime, int i)
{
	if (multime[i] == i)
	{
		return i;
	}
	multime[i] = KruskallFindMultime(multime, multime[i]);
	return multime[i];
}

void GrafPonderat::Kruskall()
{
	vector<int> indici(nrMuchii, -1);
	vector<int> multime(nrNoduri + 1, -1);
	int suma = 0;
	vector<int> muchii;

	for (int i = 0; i < nrMuchii; i++)
	{
		indici[i] = i;
	}

	for (int i = 1; i <= nrNoduri; i++)
	{
		multime[i] = i;
	}
	
	for (auto i : reprezentareGrafPonderat)
	{
		costuri.push_back(i.second);
	}
	
	sort(indici.begin(), indici.end(), KruskallComp);

	for (int i = 1; i <= nrMuchii; i++)
	{
		int indexIndici = i - 1;
		if (KruskallFindMultime(multime, reprezentareGrafPonderat[indici[indexIndici]].first.first) != KruskallFindMultime(multime, reprezentareGrafPonderat[indici[indexIndici]].first.second))
		{
			suma+=reprezentareGrafPonderat[indici[indexIndici]].second;
			KruskallReuniune(multime, reprezentareGrafPonderat[indici[indexIndici]].first.first, reprezentareGrafPonderat[indici[indexIndici]].first.second);
			muchii.push_back(indici[indexIndici]);
		}
	}
	
	fout << suma<<"\n";
	fout << muchii.size() << "\n";
	for (int i = 0; i < muchii.size(); i++)
	{
		fout << reprezentareGrafPonderat[muchii[i]].first.first << ' ' << reprezentareGrafPonderat[muchii[i]].first.second << "\n";
	}
}

Graf::Graf()
{
	nrMuchii = 0;
	nrNoduri = 0;
	orientat = false;
}

Graf::Graf(int n, int m, bool orr)
{

	nrNoduri = n;
	nrMuchii = m;
	orientat = orr;
	
}

void Graf::setListaAdiacenta()
{
	int x, y;

	listaAdiacenta.resize(nrNoduri + 1);	

	for (int i = 0; i < nrMuchii; i++)
	{
		fin >> x >> y;
		adaugaMuchie(x, y, orientat);
	}
}

void Graf::adaugaMuchie(int n1, int n2, bool orr)
{
	listaAdiacenta[n1].push_back(n2);
	if (!orr)
	{
		listaAdiacenta[n2].push_back(n1);
	}
}

void Graf::BFS(int nodStart)
{
	queue<int> q;
	vector<bool> vizitate(nrNoduri+1, false);		//vector de bool initializat cu false, in el vom retine daca nodurile au fost vizitate
	vector<int> distante(nrNoduri+1, -1);			//vector de int care retine distantele de la nodul de start, initializat cu -1

	q.push(nodStart);

	vizitate[nodStart] = true;
	distante[nodStart] = 0;

	while (!q.empty())
	{
		int nodCurent = q.front();
		q.pop();

		for (int nodVecin : listaAdiacenta[nodCurent])
		{
			if (!vizitate[nodVecin])
			{
				vizitate[nodVecin] = true;			//BFS normal, iar in if() calculam si distanta dintre nodul curent si nodul de start
				q.push(nodVecin);					//Distanta creste cu 1 de fiecare data cand trecem de la un nod la altul
				distante[nodVecin] = distante[nodCurent] + 1;
			}
		}
	}

	for (int i=1; i < distante.size(); i++)
	{
		fout << distante[i] << ' ';					//Afisam distantele
	}
}

void Graf::DFSAux(int nodStart, vector<bool>& vizitate)
{
	vizitate[nodStart] = true;
	for (int i : listaAdiacenta[nodStart])
	{
		if (!vizitate[i])					//DFS care modifica lista de noduri vizitate din metoda care rezolva problema DFS
		{
			DFSAux(i, vizitate);
		}
	}
	
}

int Graf::DFSrezolvare(int nodStart)
{
	vector<bool> vizitate(nrNoduri+1, false);
	int nrComponente = 0;
	
	for (int i = 1; i < vizitate.size(); i++)
	{
		if (!vizitate[i])
		{									//Incrementam numarul de componente pentru fiecare apel al DFSAux
			DFSAux(i, vizitate);			//Pt ca fiecare apel DFS marcheaza toate nodurile din cate o componenta conexa
			nrComponente++;					//Deci fiecare apel ne baga intr-o componenta conexa noua
		}
	}
	return nrComponente;
}

void Graf::DFSTopological(int nodStart, vector<bool>& vizitate, stack<int>& s)
{
	vizitate[nodStart] = true;
	for (int i : listaAdiacenta[nodStart])
	{
		if (!vizitate[i])
		{
			DFSTopological(i, vizitate, s);				//DFS care construieste stackul cu nodurile in ordine pentru sortarea topologica
		}												//precum ne spun si indicatiile de rezolvare de pe infoarena
	}
	s.push(nodStart);
}

void Graf::BCXAux(int nodStart, int& adancimeStart, vector<int>& adancimi, vector<int>& adancimeMinima, vector<int>& parinti, stack<int>& s, vector<vector<int>>& componenteBCX)
{
	adancimi[nodStart] = adancimeStart;					//vector care retine adancimea nodurilor in parcurgerea DFS
	adancimeMinima[nodStart] = adancimeStart;			//vector care retine cat de sus(adancime) poate un nod sa ajunga pe un drum in "jos"
	adancimeStart++;									//adancimea curenta in DFS

	for (auto vecin : listaAdiacenta[nodStart])
	{
		s.push(nodStart);								//pusham fiecare nod in stiva
		
		if (parinti[vecin] == -1)						//=> vecin nu este vizitat
		{
			parinti[vecin] = nodStart;					//parintele vecinului va fi nodul cuc are apelam functia

			BCXAux(vecin, adancimeStart, adancimi, adancimeMinima, parinti, s, componenteBCX);		//DFS

			adancimeMinima[nodStart] = min(adancimeMinima[vecin], adancimeMinima[nodStart]);		//actualizam adancimea minima a nodului curent
																									//practic verificam daca fiul sau poate ajunge la un stramos al lui
																																											
				
			if (adancimi[nodStart] <= adancimeMinima[vecin])			//nodStart == punct critic    =>    aici se termina componenta biconexa
			{
				vector<int> componenta;
				componenta.push_back(nodStart);

				vector<bool> adaugat(nrNoduri + 1, false);				//nevoie de acest vector pt ca pusham acelasi nod de mai multe ori in stiva in unele cazuri
				adaugat[nodStart] = true;								

				int nod = s.top();
				s.pop();
				componenta.push_back(nod);
				adaugat[nod] = true;

				while (nod != nodStart)						//construim componenta cu toate nodurile din stiva 
				{												
					nod = s.top();					
					s.pop();
					if (!adaugat[nod]) {
						componenta.push_back(nod);
						adaugat[nod] = true;
					}	
				}

				componenteBCX.push_back(componenta);
			}
		}
		else
		{
			adancimeMinima[nodStart] = min(adancimi[vecin], adancimeMinima[nodStart]);				//daca vecin este vizitat => adancimi[vecin] < adancimi[nodStart]
		}																							//=> actualizam adancimeaMinima[nodStart] cu minimul dintre adancimea 
																									//vecinului si adancimeaMinima a nodStart
		
	}
}

void Graf::Biconexe()
{
	vector<int> adancimi(nrNoduri + 1, -1);
	vector<int> adancimeMinima(nrNoduri + 1, -1);
	stack<int> s;
	vector<vector<int>> componente;
	vector<int> parinti(nrNoduri + 1, -1);
	int adancime = 1;
	parinti[1] = 1;
	BCXAux(1,adancime,adancimi,adancimeMinima,parinti,s, componente);

	fout << componente.size() << "\n";
	for (auto i : componente)
	{
		for (auto j : i)
		{
			fout << j << ' ';
		}
		fout << "\n";
	}

	
}

void countSort(vector<int>& v)
{
	int max = v[0];

	for (int i = 1; i < v.size(); i++)
	{
		if (v[i] > max)
		{
			max = v[i];
		}
	}

	vector<int> frecventa(max + 1, 0);
	int k = 0;

	for (int i = 0; i < v.size(); i++)
	{
		frecventa[v[i]]++;
	}

	for (int i = 0; i <= max; i++)
	{
		for (int j = 0; j < frecventa[i]; j++)
		{
			v[k++] = i;
		}

	}
}

void Graf::HavelHakimi()
{	
	int n, nod;
	vector<int> secventa;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> nod;
		secventa.push_back(nod);
	}
	while (true) {
		
		countSort(secventa);

		int deScazut = secventa[0];

		secventa.erase(secventa.begin());
		
		if (secventa.size() < deScazut)
		{
			cout << "Nu merge";
			break;
		}

		for (int i = 0; i < deScazut; i++)
		{
			secventa[i]--;
		}

		bool merge = true;
		bool nuMerge = false;

		for (int i = 0; i < secventa.size(); i++)
		{
			if (secventa[i] != 0)
			{
				merge = false;
			}

			if (secventa[i] < 0)
			{
				nuMerge = true;
				break;
			}
		}

		if (merge)
		{
			cout << "merge";
			break;
		}

		if (nuMerge)
		{
			cout << "nu merge";
			break;
		}
	}
	


}

void Graf::CTCAux(int nodStart, int& adancimeStart, vector<int>& adancimi, vector<int>& adancimeMinima, vector<bool>& peStiva, stack<int>& s, vector<vector<int>>& componenteCTC)
{
	adancimi[nodStart] = adancimeStart;					//vector care retine adancimea nodurilor in parcurgerea DFS
	adancimeMinima[nodStart] = adancimeStart;			//vector care retine cat de sus(adancime) poate un nod sa ajunga pe un drum in "jos"
	adancimeStart++;									//adancimea curenta in DFS

	s.push(nodStart);								//pusham fiecare nod in stiva

	peStiva[nodStart] = true;

	for (auto vecin : listaAdiacenta[nodStart])
	{
		if (adancimi[vecin] == -1)						//=> vecin nu este vizitat
		{
			CTCAux(vecin, adancimeStart, adancimi, adancimeMinima, peStiva, s, componenteCTC);		//DFS

			adancimeMinima[nodStart] = min(adancimeMinima[vecin], adancimeMinima[nodStart]);		//actualizam adancimea minima a nodului curent
																									//practic verificam daca fiul sau poate ajunge la un stramos al lui
		}
		else if(peStiva[vecin] == true)
		{
			adancimeMinima[nodStart] = min(adancimi[vecin], adancimeMinima[nodStart]);				//daca vecin este vizitat => adancimi[vecin] < adancimi[nodStart]
		}																							//=> actualizam adancimeaMinima[nodStart] cu minimul dintre adancimea 
																									//vecinului si adancimeaMinima a nodStart
	}

	int nod;

	if (adancimeMinima[nodStart] == adancimi[nodStart])							//varful subarborelui care formeaza CTC
	{
		vector<int>componenta;
		while (s.top() != nodStart)
		{
			nod = s.top();
			componenta.push_back(nod);						//construim componenta si o pusham in vectorul de componente
			peStiva[nod] = false;
			s.pop();
		}
		nod = s.top();
		componenta.push_back(nod);
		componenteCTC.push_back(componenta);
		peStiva[nod] = false;
		s.pop();

	}
}

void Graf::CTC()
{
	vector<int> adancimi(nrNoduri + 1, -1);
	vector<int> adancimeMinima(nrNoduri + 1, -1);
	stack<int> s;
	vector<vector<int>> componente;
	vector<bool> peStiva(nrNoduri + 1, false);
	int adancime = 1;
	for (int i = 1; i < nrNoduri + 1; i++)
	{
		if (adancimi[i] == -1)
		{
			CTCAux(i, adancime, adancimi, adancimeMinima, peStiva, s, componente);
		}
	}
	fout << componente.size() << "\n";
	for (auto i : componente)
	{
		for (auto j : i)
		{
			fout << j << ' ';
		}
		fout << "\n";
	}
}

void Graf::MuchiiAux(int nodStart, int& adancimeStart, vector<int>& adancimi, vector<int>& adancimeMinima, vector<int>& parinti, vector<bool>& vizitate, vector<vector<int>>& MuchiiCritice)
{
	adancimi[nodStart] = adancimeStart;					//vector care retine adancimea nodurilor in parcurgerea DFS
	adancimeMinima[nodStart] = adancimeStart;			//vector care retine cat de sus(adancime) poate un nod sa ajunga pe un drum in "jos"
	adancimeStart++;									//adancimea curenta in DFS
	vizitate[nodStart] = true;
	for (auto vecin : listaAdiacenta[nodStart])
	{
		if (!vizitate[vecin])						//=> vecin nu este vizitat
		{

			parinti[vecin] = nodStart;					//parintele vecinului va fi nodul cuc are apelam functia

			MuchiiAux(vecin, adancimeStart, adancimi, adancimeMinima, parinti, vizitate, MuchiiCritice);		//DFS

			adancimeMinima[nodStart] = min(adancimeMinima[vecin], adancimeMinima[nodStart]);		//actualizam adancimea minima a nodului curent
																									//practic verificam daca fiul sau poate ajunge la un stramos al lui
			
			
			if (adancimeMinima[vecin] > adancimi[nodStart])				//conditie muchie critica
			{
				vector<int> muchieCritica;
				muchieCritica.push_back(nodStart);
				muchieCritica.push_back(vecin);

				MuchiiCritice.push_back(muchieCritica);
			}

		}
			
		else if(vecin != parinti[nodStart])				
		{
			adancimeMinima[nodStart] = min(adancimi[vecin], adancimeMinima[nodStart]);				//daca vecin este vizitat => adancimi[vecin] < adancimi[nodStart]
		}																							//=> actualizam adancimeaMinima[nodStart] cu minimul dintre adancimea 
																									//vecinului si adancimeaMinima a nodStart

	}
}

void Graf::MuchiiCrit()
{
	vector<bool> vizitate(nrNoduri + 1, false);
	vector<int> adancimi(nrNoduri + 1, -1);
	vector<int> adancimeMinima(nrNoduri + 1, -1);
	vector<vector<int>> muchii;
	vector<int> parinti(nrNoduri + 1, -1);
	int adancime = 1;
	parinti[1] = 1;
	for (int i = 1; i < vizitate.size(); i++) 
	{
		if (!vizitate[i])
		{
			MuchiiAux(i, adancime, adancimi, adancimeMinima, parinti,vizitate, muchii);
		}
	}
	for (auto i : muchii)
	{
		for (auto j : i)
		{
			fout << j << ' ';
		}
		fout << "\n";
	}


}

void Graf::SortareTopologica()
{
	stack<int> s;
	vector<bool> vizitate(nrNoduri + 1, false);

	for (int i = 1; i < vizitate.size(); i++)
	{
		if (!vizitate[i])
		{
			DFSTopological(i, vizitate, s);
		}
	}

	while (!s.empty())
	{
		fout << s.top()<<' ';
		s.pop();
	}
}

int main()
{
	int n, m;
	fin >> n >> m;
	//Graf G;
	//G.HavelHakimi();

	//Graf G(n, m, true/false);	
	//G.Metoda();

	GrafPonderat G(n, m, true);
	G.setListaAdiacentaPonderata();

	G.Dijkstra(1);


	return 0;
}