Pagini recente » Cod sursa (job #1536559) | Cod sursa (job #2670471) | Cod sursa (job #2562062) | Cod sursa (job #2135434) | Cod sursa (job #2815290)
#include <iostream>
#include <fstream>
#include <stdio.h>
#include <queue>
#include <vector>
#include <utility>
#include <bits/stdc++.h>
using namespace std;
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
#define NMAX 100001
class Graf{
private:
int nrNod, nrMuch;
bool orientat;
vector<vector<int>> listaAd;
public:
Graf(int nrNoduri = 0, int nrMuchii = 0, bool eOrientat = false)
{
this->nrNod = nrNoduri;
this->nrMuch = nrMuchii;
this->orientat = eOrientat;
}
~Graf()
{
this->nrNod = 0;
this->nrMuch = 0;
listaAd.clear();
}
void set_nrNod(int &);
void set_nrMuch(int &);
int get_nrNod();
int get_nrMuch();
void citire_muchii();
void citire_matrice(vector<vector<int>>&);
void citire_capacitate(vector<vector<int>>&, vector<vector<int>>&);
//lab 3
void bfs(int);
//lab 4
void init();
int reprez(int);
void unite(int,int);
void apm_kruskall();
void dijkstra(int);
bool bellmanford(int);
void disjoint();
//lab 5
int darb(int,int&);
void roy_floyd(vector<vector<int>>&);
bool bfs_flow(int, int, vector<int>&, vector<vector<int>>&);
int max_flow(vector<int>&, vector<vector<int>>&);
};
void Graf::set_nrNod(int &n) {nrNod = n;}
void Graf::set_nrMuch(int &m) {nrMuch = m;}
int Graf::get_nrNod() {return nrNod;}
int Graf::get_nrMuch() {return nrMuch;}
void Graf::citire_muchii()
{
int x, y;
listaAd.resize(NMAX+1);
for(int i = 0; i < nrMuch; i++)
{
fin >> x >> y;
listaAd[x].push_back(y);
listaAd[y].push_back(x);
}
}
void Graf::citire_matrice(vector<vector<int>>& A)
{
int x;
A.resize(nrNod+1);
for(int i = 1; i <= nrNod; i++)
{
A[i].resize(nrNod+1);
for(int j = 1; j <= nrNod; j++)
{
fin >> x;
A[i][j] = x;
}
}
}
void Graf::citire_capacitate(vector<vector<int>>& c, vector<vector<int>>& rc)
{
int x,y,cap;
listaAd.resize(NMAX+1);
c.resize(nrNod+1);
rc.resize(nrNod+1);
for(int i = 1; i <= nrNod; i++)
{
c[i].resize(nrNod+1);
rc[i].resize(nrNod+1);
}
for(int i = 0; i < nrMuch; i++)
{
fin >> x >> y >> cap;
listaAd[x].push_back(y);
c[x][y] = cap;
rc[x][y] = cap;
}
}
int Graf::darb(int nod, int& diam)
{
int viz[NMAX+1], d[NMAX+1], cap;
queue<int> q;
for(int i = 1; i <= NMAX; i++)
{
d[i] = 0;
viz[i] = 0;
}
q.push(nod);
d[nod] = 1;
viz[nod] = 1;
int x;
while(!q.empty())
{
x = q.front();
for(auto vecin:listaAd[x])
{
if(!viz[vecin])
{
q.push(vecin);
viz[vecin] = 1;
d[vecin] = d[x] + 1;
diam = d[vecin];
cap = vecin;
}
}
q.pop();
}
return cap;
}
void Graf::roy_floyd(vector<vector<int>>& A)
{
for(int k = 1; k <= nrNod; k++)
for(int i = 1; i <= nrNod; i++)
for(int j = 1; j <= nrNod; j++)
if((i!=j) && A[i][k] && A[k][j] && (A[i][j] > A[i][k] + A[k][j] || !A[i][j]))
A[i][j] = A[i][k] + A[k][j];
}
bool Graf::bfs_flow(int s, int f, vector<int>& t, vector<vector<int>>& rc)
{
int x;
int viz[nrNod+1];
for(int i = 1; i <= nrNod; i++)
viz[i] = 0;
queue<int> q;
q.push(s);
viz[s] = 1;
t[s] = -1;
while(!q.empty())
{
x = q.front();
q.pop();
for(auto vecin:listaAd[x])
{
if (!viz[vecin] && rc[x][vecin] > 0)
{
if (vecin == f)
{
t[vecin] = x;
return true;
}
q.push(vecin);
t[vecin] = x;
viz[vecin] = 1;
}
}
}
return false;
}
int Graf::max_flow(vector<int>& t, vector<vector<int>>& rc)
{
int rasp = 0, p, nod;
int path_flow;
while(bfs_flow(1, nrNod, t, rc))
{
path_flow = INT_MAX;
for(nod = nrNod; nod != 1; nod = t[nod])
{
p = t[nod];
path_flow = min(path_flow, rc[p][nod]);
}
for(nod = nrNod; nod != 1; nod = t[nod])
{
p = t[nod];
rc[p][nod] -= path_flow;
rc[nod][p] += path_flow;
}
rasp += path_flow;
}
return rasp;
}
void darb_infoarena()
{
int n;
fin >> n;
Graf G(0,n,false);
int cap, diam;
G.citire_muchii();
cap = G.darb(1, diam);
G.darb(cap, diam);
fout << diam;
}
void roy_floyd_infoarena()
{
int n;
vector<vector<int>> A;
fin >> n;
Graf G(n,0,true);
G.citire_matrice(A);
G.roy_floyd(A);
for(int i = 1; i <= n; i++)
{
for(int j = 1; j <= n; j++)
{
fout << A[i][j] << ' ';
}
fout << endl;
}
}
void maxflow_infoarena()
{
int n, m, rasp = 0;
vector<vector<int>> c,rc;
vector<int> t;
fin >> n >> m;
t.resize(n+1);
Graf G (n,m,true);
G.citire_capacitate(c,rc);
rasp = G.max_flow(t,rc);
fout << rasp;
}
int main()
{
maxflow_infoarena();
fin.close();
fout.close();
return 0;
}