TLE求助
查看原帖
TLE求助
602803
_qingshu_楼主2024/1/18 21:04

这费用流怎么写一次挂一次qwq

除了 #1 全挂了。

#include<bits/stdc++.h>
#define inf LONG_LONG_MAX
using namespace std;
int n,m,k,s,t;
struct edge{
    int u,v,c,w;
}in[5200010];
struct node{
    int to,nxt;
    long long val,spendf;
}e[5200010];
long long head[5200010],tot=1,maxval,maxspendf;
inline void add(int u,int v,long long val,long long spendf){
    e[++tot].to=v;
    e[tot].nxt=head[u];
    e[tot].val=val;
    e[tot].spendf=spendf;
    head[u]=tot;

    e[++tot].to=u;
    e[tot].nxt=head[v];
    e[tot].val=0;
    e[tot].spendf=-spendf;
    head[v]=tot;
}
long long dis[5200010],vis[5200010],now[5200010];
inline bool spfa(){
    queue<int>q;
    q.push(s);
    for(int i=0;i<=t;i++)dis[i]=inf;
    dis[s]=0;
    now[s]=head[s];
    while(!q.empty()){
        int x=q.front();
        q.pop();
        vis[x]=0;
        for(int i=head[x];i;i=e[i].nxt){
            int v=e[i].to;
            if(e[i].val&&dis[v]>dis[x]+e[i].spendf){
                dis[v]=dis[x]+e[i].spendf;
                now[v]=head[v];
                if(!vis[v]){
                    q.push(v);
                    vis[v]=1;
                }
            }
        }
    }
    return dis[t]!=inf;
}
inline long long dfs(int x,long long sum){
    if(x==t)return sum;
    long long res=0,i;
    vis[x]=1;
    for(i=head[x];i;i=e[i].nxt){
        now[x]=i;
        int v=e[i].to;
        if(e[i].val&&!vis[v]&&dis[v]==dis[x]+e[i].spendf){
            long long k=dfs(v,min(sum,e[i].val));
            e[i].val-=k;
            e[i^1].val+=k;
            sum-=k;
            res+=k;
            maxspendf+=k*e[i].spendf;
        }
    }
    vis[x]=0;
    return res;
}
inline void dinic(){
    while(spfa()){
        maxval+=dfs(s,inf);
    }
}
int main(){
    cin>>n>>m>>k;s=1,t=n;
    for(int i=1;i<=m;i++){
        cin>>in[i].u>>in[i].v>>in[i].c>>in[i].w;
        add(in[i].u,in[i].v,in[i].c,0);
    }
    dinic();
    cout<<maxval<<" ";
    s=0;
    add(s,1,k,0);
    for(int i=1;i<=m;i++){
        add(in[i].u,in[i].v,inf,in[i].w);
    }
    dinic();
    cout<<maxspendf;
}
2024/1/18 21:04
加载中...