topcoder FoxAndCity

最新推荐文章于 2019-09-12 21:24:33 发布

sdfzyhx

最新推荐文章于 2019-09-12 21:24:33 发布

阅读量392

点赞数

分类专栏：图论 topcoder 文章标签：最小割

本文链接：https://blog.csdn.net/sdfzyhx/article/details/70780701

版权

图论同时被 2 个专栏收录

177 篇文章 1 订阅

订阅专栏

topcoder

3 篇文章 0 订阅

订阅专栏

可以看成给每个点随便选权值，有相邻边的点权值相差不超过 1 <script type="math/tex" id="MathJax-Element-25">1</script>。因为原图是联通的，每种满足要求的方案都一定是合法的。这样建图跑最小割就行了，参见【HNOI2013】bzoj3144 切糕。

#include<algorithm>
#include<vector>
#include<string>
using namespace std;
const int maxn=200010,s=20005,t=20006,oo=0x3f3f3f3f;
class FoxAndCity
{
    int fir[maxn],ne[maxn],to[maxn],w[maxn],
    que[maxn],dep[maxn],cur[maxn],id[45][45],
    n,num,tot;
private:
    void add(int u,int v,int x)
    {
        num++;
        ne[num<<1]=fir[u];
        fir[u]=num<<1;
        to[num<<1]=v;
        w[num<<1]=x;
        ne[num<<1|1]=fir[v];
        fir[v]=num<<1|1;
        to[num<<1|1]=u;
        w[num<<1|1]=0;
    }
    int bfs()
    {
        int hd=1,tl=1,u,v;
        for (int i=1;i<=tot;i++) dep[i]=0;
        dep[t]=0;
        que[dep[s]=1]=s;
        while (hd<=tl)
        {
            u=que[hd++];
            cur[u]=fir[u];
            for (int i=fir[u];i;i=ne[i])
                if (w[i]&&!dep[v=to[i]])
                {
                    dep[v]=dep[u]+1;
                    que[++tl]=v;
                }
        }
        return dep[t];
    }
    int dfs(int u,int lim)
    {
        if (u==t) return lim;
        int ret=0,v,x;
        for (int &i=cur[u];i&&ret<lim;i=ne[i])
            if (w[i]&&dep[v=to[i]]==dep[u]+1)
            {
                x=dfs(v,min(w[i],lim-ret));
                ret+=x;
                w[i]-=x;
                w[i^1]+=x;
            }
        return ret;
    }
public:
    int minimalCost(vector<string> linked,vector<int> want)
    {
        int ans=0;
        n=want.size();
        tot=num=0;
        for (int i=0;i<n;i++)
            for (int j=0;j<=n;j++)
                id[i][j]=++tot;
        for (int i=1;i<=tot;i++) fir[i]=0;
        fir[s]=fir[t]=0;
        add(s,id[0][0],oo);
        add(id[0][n],t,oo);
        for (int i=1;i<n;i++) add(id[0][i],id[0][i+1],oo);
        for (int i=1;i<n;i++)
        {
            add(s,id[i][0],oo);
            add(id[i][0],id[i][1],oo);
            add(id[i][n],t,oo);
            for (int j=want[i]-1;j>=1;j--)
                add(id[i][j],id[i][j+1],(want[i]-j)*(want[i]-j));
            for (int j=want[i]+1;j<n;j++)
                add(id[i][j],id[i][j+1],(j-want[i])*(j-want[i]));
        }
        for (int i=0;i<n;i++)
            for (int j=i+1;j<n;j++)
                if (linked[i][j]=='Y')
                    for (int k=0;k<n;k++)
                    {
                        add(id[i][k+1],id[j][k],oo);
                        add(id[j][k+1],id[i][k],oo);
                    }
        while (bfs()) ans+=dfs(s,oo);
        return ans;
    }
};