将原来就有的边转化为|dis(u)-dis(v)|<=1的约束,然后建图最小割,一个点拆n个点的从S到T的链表示最终离0号点距离的情况,然后在T中的点pt(i,x)表示满足dis(i)<=x,在S中的点表示不满足,而要保证这些限制就从T到S每条边连inf反向边,这样显然可以保证靠T的边被连续取到,即满足<=x也满足<=x+1...,之后把原图的边转化为<=1约束在u,v每种情况间连inf边表示一种不可行另一种也不可行即可。
代码(被魔改过的题):
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=65,M=1e6;
const ll inf=1e18;
template<class T>
void rd(T &x)
{
char c=getchar();x=0;bool f=0;
while(!isdigit(c))f|=(c=='-'),c=getchar();
while(isdigit(c))x=x*10+c-48,c=getchar();
if(f)x=-x;
}
struct Edge{
int u,v;
}e[N*N];
ll ans=0;
int n,m,a[N],S,T,pnum=0;
int hd[M],nxt[M],to[M],cur[M],tot=-1,pt[N][N],dep[N*N];
ll las[M];
vector<int>mp[N];
ll calc(ll x,ll y)
{return (ll)(10LL*log(233LL*(x-y)*(x-y)+1));}
void add(int u,int v,ll la)
{
nxt[++tot]=hd[u],to[tot]=v,las[tot]=la,hd[u]=tot;
nxt[++tot]=hd[v],to[tot]=u,las[tot]=0,hd[v]=tot;
}
bool bfs()
{
int nw;
queue<int>q;
memset(dep,0,sizeof dep);
dep[S]=1,q.push(S);
while(!q.empty())
{
nw=q.front(),q.pop();
for(int i=hd[nw],v;~i;i=nxt[i])
{
v=to[i];
if(las[i]&&!dep[v])
{
dep[v]=dep[nw]+1;
q.push(v);
}
}
}
return dep[T];
}
ll dfs(int x,ll flow)
{
if(x==T)return flow;
ll tmp;
for(int &i=cur[x],v;~i;i=nxt[i])
{
v=to[i];
if(las[i]&&dep[v]==dep[x]+1)
{
tmp=dfs(v,min(flow,las[i]));
if(tmp)
{
las[i]-=tmp;
las[i^1]+=tmp;
return tmp;
}
}
}
return 0;
}
void dinic()
{
ll res=0,tmp;
while(bfs())
{
for(int i=1;i<=pnum;i++)cur[i]=hd[i];
while(tmp=dfs(S,inf))res+=tmp;
}
printf("%lld\n",res);
}
void build()
{
S=++pnum,T=++pnum;
memset(hd,-1,sizeof hd);
for(int i=1;i<=n;i++)
{
for(int j=0;j<=n-1;j++)
pt[i][j]=++pnum;
if(i==1)
{
add(S,pt[1][0],calc(0,a[1]));
for(int j=0;j<n-1;j++)
add(pt[1][j],pt[1][j+1],inf);
add(pt[1][n-1],T,inf);
}
else
{
add(S,pt[i][0],inf);
for(int j=0;j<n-1;j++)
add(pt[i][j],pt[i][j+1],calc(j+1,a[i]));
add(pt[i][n-1],T,inf);
}
add(T,pt[i][n-1],inf);
for(int j=n-1;j>=1;j--)
add(pt[i][j],pt[i][j-1],inf);
}
for(int i=1,u,v;i<=m;i++)
{
u=e[i].u,v=e[i].v;
for(int j=1;j<n;j++)
{
add(pt[u][j],pt[v][j-1],inf);
add(pt[v][j],pt[u][j-1],inf);
}
}
dinic();
}
int main()
{
rd(n),rd(m);
for(int i=1;i<=n;i++)
rd(a[i]);
for(int i=1;i<=m;i++)
rd(e[i].u),rd(e[i].v);
build();
}