Cod sursa(job #2603201)

Utilizator mihaimodiMihai Modi mihaimodi Data 18 aprilie 2020 18:40:13
Problema Cc Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 2.08 kb
#include <fstream>
#include <queue>
#include <vector>
#define inf 2000000000
using namespace std;
ifstream fin("cc.in");
ofstream fout("cc.out");
vector<int> g[205];
queue<int> q;
int tata[205];
int f[205][205],c[205][205],cost[205][205];
int d[205];
bool viz[205];
int n,m,x,y;
long long sol;
int bfs()
{
    q.empty();
    q.push(0);
    for(int i=0;i<=2*n+1;i++)
    {
        tata[i]=-1;
        viz[i]=0;
        d[i]=inf;
    }
    d[0]=0;
    viz[0]=1;
    while(!q.empty())
    {
        int nod=q.front();
        viz[nod]=0;
        q.pop();
        for(int i=0;i<g[nod].size();i++)
        {
            int nou=g[nod][i];
            if(c[nod][nou]-f[nod][nou]>0)
            {
                int dist=d[nod]+cost[nod][nou];
                if(dist<d[nou])
                {
                    if(viz[nou]==0)
                    {
                        viz[nou]=1;
                        q.push(nou);
                    }
                    d[nou]=dist;
                    tata[nou]=nod;
                }
            }
        }
    }
    return d[2*n+1]!=inf;
}
/*
int minim(int nod)
{
    int Min=inf;
    while(nod!=0)
    {
        int nou=tata[nod];
        Min=min(Min,c[nou][nod]-f[nou][nod]);
        nod=nou;
    }
    return Min;
}
*/
void add(int val,int nod)
{
    while(nod!=-1)
    {
        int nou=tata[nod];
        f[nou][nod]+=val;
        f[nod][nou]-=val;
        nod=nou;
    }
}
int main()
{
    fin>>n;
    for(int i=1;i<=n;i++)
        for(int j=n+1;j<=2*n;j++)
        {
            g[i].push_back(j);
            g[j].push_back(i);
            fin>>cost[i][j];
            cost[j][i]=-cost[i][j];
            c[i][j]=1;
        }
    for(int i=1;i<=n;i++)
    {
        g[0].push_back(i);
        g[i].push_back(0);
        c[0][i]=1;
    }
    for(int i=n+1;i<=2*n;i++)
    {
        g[i].push_back(2*n+1);
        g[2*n+1].push_back(i);
        c[i][2*n+1]=1;
    }
    while(bfs()==1)
    {
        add(1,2*n+1);
        sol+=d[2*n+1];
    }
    fout<<sol;
    return 0;
}