Cod sursa(job #1159565)

Utilizator Dddarius95Darius-Florentin Neatu Dddarius95 Data 29 martie 2014 18:30:24
Problema Cuplaj maxim de cost minim Scor 20
Compilator cpp Status done
Runda Arhiva educationala Marime 2.61 kb

// O(N*M^2*LogN)
#include <fstream>
#include <vector>
#include <queue>
#include <cstring>
#define Nmax 605
#define oo 2000000000
#define pb push_back
#define x first
#define y second
using namespace std;
ifstream f("cmcm.in");
ofstream g("cmcm.out");

int L,R,N,M,Source,Sink,x,y,c,z,MaxFlow,CostMin;
int cap[Nmax][Nmax],cost[Nmax][Nmax],flow[Nmax][Nmax],nr[Nmax][Nmax],ante[Nmax],d[Nmax],old_d[Nmax],real_d[Nmax];
vector < int > G[Nmax],sol;

typedef pair<int,int> PP;
priority_queue < PP  , vector < PP > , greater<PP> > pq;

inline void ReadInput()
{
    f>>L>>R>>M;  Source=0 , Sink=L+R+1;
    N=L+R+1;
    for(int i=1;i<=M;++i)
        {
            f>>x>>y>>z;
            y+=L;
            G[x].pb(y) , G[y].pb(x);
            cap[x][y]=1;
            cost[x][y]=z , cost[y][x]=-z;
            nr[x][y]=nr[y][x]=i;

            G[0].pb(x) , G[x].pb(0);
            cap[0][x]=1;

            G[y].pb(Sink) , G[Sink].pb(y);
            cap[y][Sink]=1;
        }
}

int Dijkstra(int Source)
{
    for(int i=0;i<=N;++i)d[i]=oo,ante[i]=oo;
    d[Source]=0 , ante[Source]=Source;
    for(pq.push(PP(0,Source)); !pq.empty();)
    {
        int  val=pq.top().x,node=pq.top().y;
        pq.pop();
        if(d[node]<val)continue;
        for(vector<int>::iterator it=G[node].begin();it!=G[node].end();++it)
            if(flow[node][*it]<cap[node][*it])
            {
                int aux=d[node]+cost[node][*it]+old_d[node]-old_d[*it];
                if(aux<d[*it])
                {
                    d[*it]=aux , ante[*it] = node;
                    real_d[*it]=real_d[node]+cost[node][*it];
                    pq.push( PP(d[*it] , *it) );
                }
            }
    }
    for(int i=0;i<=N;++i)old_d[i]=real_d[i];
    return (d[Sink]!=oo) ;
}

void GetFMCM()
{
    for( ; Dijkstra(Source) ; )
    {
        int MinFlow=oo;
        for(int i=Sink; i!=Source ; i=ante[i])
            MinFlow=min(MinFlow,cap[ante[i]][i]-flow[ante[i]][i]);
        if(MinFlow)
        {
            int muchie=0;
            for(int i=Sink; i!=Source ; i=ante[i])
            {
                flow[ante[i]][i]+=MinFlow;
                flow[i][ante[i]]-=MinFlow;
                ++muchie;
                if(muchie==2)sol.pb(nr[ante[i]][i]);
            }
            MaxFlow+=MinFlow;
            CostMin+=MinFlow*real_d[Sink];
        }
    }
    g<<sol.size()<<' '<<CostMin<<'\n';
    for(vector<int>::iterator it=sol.begin();it!=sol.end();++it)
        g<<*it<<' ';
    g<<'\n';
}

int main()
{
    ReadInput();
    GetFMCM();
    f.close();g.close();
    return 0;
}