Pagini recente » Cod sursa (job #1776120) | Cod sursa (job #583629) | Cod sursa (job #1271437) | Cod sursa (job #2683050) | Cod sursa (job #2669666)
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp> // Common file
#include <ext/pb_ds/tree_policy.hpp> // Including tree_order_statistics_node_update
#define MIN(a, b) (((a) < (b)) ? (a) : (b))
#define MAX(a, b) (((a) < (b)) ? (b) : (a))
#define pb(a) push_back(a)
//#pragma GCC optimize("O3")
using namespace std;
using namespace __gnu_pbds;
auto random_address = [] { char *p = new char; delete p; return uint64_t(p); };
const uint64_t SEED = chrono::steady_clock::now().time_since_epoch().count() * (random_address() | 1);
mt19937_64 rng(SEED);
typedef tree<
int,
null_type,
less<int>,
rb_tree_tag,
tree_order_statistics_node_update>
ordered_set;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ld;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
ifstream fin("maxflow.in");
ofstream fout("maxflow.out");
ll n,m,flux[1001][1001],cap[1001][1001],ans,nrways;
vector<pll> muchii[1001];
deque<pair<pll,ll>> path[10001];
vector<int> sols;
const ll INF=1000000000;
bool use[1001];
void dfs(int nod,int nr)
{
use[nod]=1;
for(auto it:muchii[nod])
{
int i=it.first;
int c=it.second;
if(flux[nod][i]<cap[nod][i])
{
if(i==n)
{
if(nod==1)
{
nrways++;
nr=nrways;
}
path[nr].push_back({{nod,i},c});
sols.push_back(nr);
use[n]=1;
}
else
{
if(use[i]==0)
{
if(nod==1)
{
nrways++;
nr=nrways;
}
path[nr].push_back({{nod,i},c});
dfs(i,nr);
}
}
}
}
}
int main()
{
ios_base::sync_with_stdio(false);
fin.tie(0);
fout.tie(0);
fin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
cap[i][j]=-1;
for(int i=1;i<=m;i++)
{
int a,b,c;
fin>>a>>b>>c;
muchii[a].pb(make_pair(b,c));
cap[a][b]=c;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
if(cap[i][j]!=-1)
if(cap[j][i]==-1)
{
cap[j][i]=0;
muchii[j].pb(make_pair(i,0));
}
while(true)
{
for(int i=1;i<=nrways;i++)
path[i].clear();
nrways=0;
sols.clear();
memset(use,0,sizeof(use));
dfs(1,0);
if(!use[n])
break;
for(auto p:sols)
{
ll val=INF;
for(auto i:path[p])
{
ll nod1=i.first.first;
ll nod2=i.first.second;
ll dif=cap[nod1][nod2]-flux[nod1][nod2];
val=min(val,dif);
}
if(val==INF||val<=0)
continue;
ans+=val;
for(auto i:path[p])
{
ll nod1=i.first.first;
ll nod2=i.first.second;
flux[nod1][nod2]+=val;
flux[nod2][nod1]-=val;
}
}
}
fout<<ans;
return 0;
}