计蒜客 腾讯狼人杀（困难） 最大权密度子图变种

题目链接点这里

历经一个星期，，终于做好了。。

《最小割模型在信息学竞赛中的应用》这篇论文必看啊。。。

#include<iostream>
#include<cstdio>
#include<math.h>
#include<algorithm>
#include<map>
#include<set>
#include<bitset>
#include<stack>
#include<queue>
#include<string.h>
#include<cstring>
#include<vector>
#include<time.h>
#include<stdlib.h>
using namespace std;
#define INF 0x3f3f3f3f
#define INFLL 0x3f3f3f3f3f3f3f3f
#define FIN freopen("input.txt","r",stdin)
#define mem(x,y) memset(x,y,sizeof(x))
typedef unsigned long long ULL;
typedef long long LL;
#define fuck(x) cout<<"q"<<endl;
#define MX 444
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<pair<int,int>,int> PIII;
typedef pair<int,int> PII;
const double eps=1e-8;
int n,m;
double G[MX][MX];
int w[MX];
int head[MX],edge_cnt;
struct Edge
{
int nxt,to;
double cap;
Edge() {}
Edge(int nxt,int to,double cap):nxt(nxt),to(to),cap(cap) {}
} E[666666];
void edge_init()
{
mem(head,-1);
edge_cnt=0;
}
void edge_add(int u,int v,double cap)
{
E[edge_cnt]=Edge(head[u],v,cap);
head[u]=edge_cnt++;
}
//正向边和反向边一定要一起建立！！，因为用到i^1
bool vis[MX];//注意点数！！
int d[MX],cur[MX];//注意点数！！
bool BFS(int s,int t)
{
mem(vis,0);
queue<int> Q;
Q.push(s);
vis[s]=1;
d[s]=0;
d[t]=-1;
while(!Q.empty())
{
//fuck()
int u=Q.front();
Q.pop();
for(int i=head[u]; ~i; i=E[i].nxt)
{
int v=E[i].to;
if(vis[v]||E[i].cap==0) continue;
d[v]=d[u]+1;
vis[v]=1;
Q.push(v);
}
}
return d[t]!=-1;
}
double DFS(int x,int t,double a)
{
if(x==t||a==0) return a;
double flow=0,f;
for(int &i=cur[x]; ~i; i=E[i].nxt)
{
int v=E[i].to;
if(d[v]==d[x]+1&&(f=DFS(v,t,min(a,E[i].cap))))
{
E[i].cap-=f;
E[i^1].cap+=f;
flow+=f;
a-=f;
if(a==0) break;
}
}
return flow;
}
double Dinic(int s,int t)
{
double flow=0;
while(BFS(s,t))
{

memcpy(cur,head,sizeof(head));
flow+=DFS(s,t,INF);
}
return flow;
}
double du[MX];
double check(double g)
{
edge_init();
int s=0,t=n+1;
double U=0;
for(int i=1; i<=n; i++)
{
du[i]=4*n*g-2*g;
for(int j=1; j<=n; j++)
if(i!=j)du[i]-=G[i][j]+2*g;
U=min(U,du[i]);
}
U=fabs(U);
for(int i=1; i<=n; i++)
for(int j=1; j<i; j++)
{
edge_add(i,j,G[i][j]+2*g);
edge_add(j,i,G[i][j]+2*g);
}

for(int i=1; i<=n; i++)
{
edge_add(s,i,U);
edge_add(i,s,0);
edge_add(i,t,du[i]+U);
edge_add(t,i,0);
}
for(int i=1; i<=w[0]; i++) edge_add(s,w[i],INF),edge_add(w[i],s,0);
double ans=Dinic(s,t)-U*n;
return ans;
}
int main()
{
FIN;
while(cin>>n>>m)
{
double l=0,r=0;
for(int i=1; i<=n; i++)
for(int j=1; j<=n; j++)G[i][j]=0;
for(int i=1; i<=m; i++)
{
int u,v,dist;
scanf("%d%d%d",&u,&v,&dist);
r+=dist;
if(u>v)swap(u,v);
G[u][v]=G[v][u]=1.0*dist;
}
w[0]=0;
for(int i=1; i<=n; i++)
{
int x;
scanf("%d",&x);
if(x)w[++w[0]]=i;
}
while(fabs(l-r)>eps)
{
double mid=(l+r)/2;
if(check(mid)<0)l=mid;
else r=mid;
}
printf("%.6f\n",l);
}
return 0;
}