洛谷P4542 [ZJOI2011]营救皮卡丘
思路:
定义
d
i
s
[
u
]
[
v
]
(
u
>
v
)
dis[u][v](u>v)
dis[u][v](u>v)为
u
u
u到
v
v
v且不经过编号大于
v
v
v节点的最短路。用最短路可以求得。然后建图跑费用流。
建图:
拆点,把
i
i
i拆成
i
,
i
′
i,i^{'}
i,i′。
1、
S
S
S到
0
0
0建
(
S
,
0
,
k
,
0
)
(S,0,k,0)
(S,0,k,0)。
2、
S
S
S到节点
i
i
i建
(
S
,
i
,
1
,
0
)
(S,i,1,0)
(S,i,1,0)。
3、节点
i
′
i^{'}
i′到
T
T
T建
(
i
′
,
T
,
1
,
0
)
(i^{'},T,1,0)
(i′,T,1,0)。
4、
i
i
i到
j
′
(
i
<
j
,
d
i
s
i
,
j
≠
i
n
f
)
j^{'}(i<j,dis_{i,j}\neq inf)
j′(i<j,disi,j=inf)建
(
i
,
j
′
,
i
n
f
,
d
i
s
i
,
j
)
(i,j^{'},inf,dis_{i,j})
(i,j′,inf,disi,j)。
代码:
#include<bits/stdc++.h>
#define pii pair<int,int>
#define ll long long
#define cl(x,y) memset(x,y,sizeof(x))
#define loop(x,y,z) for(x=y;x<=z;x++)
#define reve(x,y,z) for(x=y;x>=z;x--)
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define all(x) x.begin(),x.end()
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define INF 1e18
const int N=1e4+10;
const int M=1e6+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
using namespace std;
struct edge
{
int u,v,f,w;
}e[M];
int head[N],len=1,dis[N],vis[N];
int maxflow,mincost;
int pre[N],path[N];
void add(int u,int v,int f,int w)
{
e[++len]={head[u],v,f,w};
head[u]=len;
}
void inc(int u,int v,int f,int w)
{
add(u,v,f,w);
add(v,u,0,-w);
}
int spfa(int s,int t)
{
cl(dis,inf);
cl(vis,0);
cl(pre,-1);
dis[s]=0;
vis[s]=1;
queue<int>q;
q.push(s);
while(!q.empty())
{
int u=q.front(),i;
q.pop();
vis[u]=0;
for(i=head[u];i;i=e[i].u)
{
int v=e[i].v,w=e[i].w;
if(e[i].f && dis[v]>dis[u]+w)
{
dis[v]=dis[u]+w;
pre[v]=u;
path[v]=i;
if(!vis[v])
{
q.push(v);
vis[v]=1;
}
}
}
}
return pre[t]!=-1;
}
void ek(int s,int t)
{
while(spfa(s,t))
{
int mi=inf,i;
for(i=t;i!=s;i=pre[i])
mi=min(mi,e[path[i]].f);
for(i=t;i!=s;i=pre[i])
{
e[path[i]].f-=mi;
e[path[i]^1].f+=mi;
}
maxflow+=mi;
mincost+=dis[t]*mi;
}
}
int d[210][210];
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,m,p,i,j,k;
cin>>n>>m>>p;
n++;
int s=0,t=2*n+1;
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
d[i][j]=(i==j?0:inf);
for(i=1;i<=m;i++)
{
int u,v,w;
cin>>u>>v>>w;
u++;
v++;
d[u][v]=min(d[u][v],w);
d[v][u]=min(d[v][u],w);
}
for(k=1;k<=n;k++)
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
if(k<max(i,j))
d[i][j]=min(d[i][j],d[i][k]+d[k][j]);
for(i=1;i<=n;i++)
{
if(i==1)
inc(s,i,p,0);
else
inc(s,i,1,0);
inc(i+n,t,1,0);
}
for(i=1;i<=n;i++)
for(j=i+1;j<=n;j++)
if(d[i][j]!=inf)
inc(i,j+n,inf,d[i][j]);
ek(s,t);
cout<<mincost<<endl;
return 0;
}