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

题目链接点这里

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

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

#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;
}