Pagini recente » Cod sursa (job #657458) | Cod sursa (job #2855957) | Cod sursa (job #1559865) | Cod sursa (job #2526841) | Cod sursa (job #3185728)
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#include <climits>
using namespace std;
const int MAXN = 202; // N * 2 + 2 (doubling each city node + source and sink)
int capacity[MAXN][MAXN];
int flow[MAXN][MAXN];
int parent[MAXN];
bool visited[MAXN];
int N;
bool bfs(int s, int t)
{
memset(visited, 0, sizeof(visited));
queue<int> q;
q.push(s);
visited[s] = true;
parent[s] = -1;
while (!q.empty())
{
int u = q.front();
q.pop();
for (int v = 0; v < 2 * N + 2; ++v)
{
if (!visited[v] && capacity[u][v] - flow[u][v] > 0)
{
q.push(v);
parent[v] = u;
visited[v] = true;
}
}
}
return visited[t];
}
int fordFulkerson(int s, int t)
{
int max_flow = 0;
while (bfs(s, t))
{
int path_flow = INT_MAX;
for (int v = t; v != s; v = parent[v])
{
int u = parent[v];
path_flow = min(path_flow, capacity[u][v] - flow[u][v]);
}
for (int v = t; v != s; v = parent[v])
{
int u = parent[v];
flow[u][v] += path_flow;
flow[v][u] -= path_flow;
}
max_flow += path_flow;
}
return max_flow;
}
int main()
{
ifstream fin("harta.in");
ofstream fout("harta.out");
fin >> N;
int total_in_deg = 0, total_out_deg = 0;
vector<pair<int, int>> cities(N);
// Reading city data
for (int i = 0; i < N; ++i)
{
fin >> cities[i].first >> cities[i].second;
total_out_deg += cities[i].first;
total_in_deg += cities[i].second;
}
// Check if solution is possible
if (total_in_deg != total_out_deg)
{
fout << "No solution" << endl;
fin.close();
fout.close();
return 0;
}
int source = 2 * N, sink = 2 * N + 1;
// Constructing the graph
for (int i = 0; i < N; ++i)
{
capacity[source][i] = cities[i].first; // From source to out nodes
capacity[i + N][sink] = cities[i].second; // From in nodes to sink
for (int j = 0; j < N; ++j)
{
if (i != j)
{
capacity[i][j + N] = 1; // From out nodes to in nodes
}
}
}
// Calculating the max flow
int max_flow = fordFulkerson(source, sink);
// Check if max flow equals total degrees
if (max_flow != total_in_deg)
{
fout << "No solution" << endl;
fin.close();
fout.close();
return 0;
}
// Extracting the roads from the flow
fout << max_flow << endl;
for (int i = 0; i < N; ++i)
{
for (int j = 0; j < N; ++j)
{
if (flow[i][j + N] > 0)
{
fout << i + 1 << " " << j + 1 << endl;
}
}
}
fin.close();
fout.close();
return 0;
}