bzoj千题计划265：bzoj4873: [六省联考2017]寿司餐厅

#include<queue>
#include<cstdio>
#include<iostream>
 
using namespace std;
 
#define N 10201
#define M 30401
 
const int inf=2e9;
 
#define min(x,y) ((x)<(y) ? (x) : (y))
 
int ty[101];
int d[101][101];
 
int n,m;
 
int id1[101][101],id2[101],id3[1001];
bool vis[1001];
 
int ans;
 
int src,decc;
int tot=1,front[N],to[M<<1],nxt[M<<1],cap[M<<1];
int lev[N],cur[N];
 
queue<int>q;
 
void read(int &x)
{
    x=0; int f=1; char c=getchar();
    while(!isdigit(c)) { if(c==‘-‘) f=-1; c=getchar(); }
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar(); }
    x*=f;
}
 
void init()
{
    read(n); read(m);
    for(int i=1;i<=n;++i) read(ty[i]);
    for(int i=1;i<=n;++i)
        for(int j=i;j<=n;++j)
        {
            read(d[i][j]);
            if(d[i][j]>0) ans+=d[i][j];
        }
}
 
void add(int u,int v,int w)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; cap[tot]=w;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; cap[tot]=0;
}
 
void build()
{
    int num=0;
    for(int i=1;i<=n;++i)
        for(int j=i;j<=n;++j)
            id1[i][j]=++num;
    for(int i=1;i<=n;++i) id2[i]=++num;
    for(int i=1;i<=n;++i)
        if(!vis[ty[i]])
        {
            vis[ty[i]]=true;
            id3[ty[i]]=++num;
        }
    decc=num+1;
    for(int i=1;i<=n;++i)
        for(int j=i;j<=n;++j)
        {
            if(d[i][j]>0) add(src,id1[i][j],d[i][j]);
            else add(id1[i][j],decc,-d[i][j]);
            if(i!=j)
            {
                add(id1[i][j],id1[i+1][j],inf);
                add(id1[i][j],id1[i][j-1],inf);
            }
        }
    for(int i=1;i<=n;++i) add(id1[i][i],id2[i],inf);
    for(int i=1;i<=n;++i)
    {
        add(id2[i],id3[ty[i]],inf);
        add(id2[i],decc,ty[i]);
    }
    if(m)
    for(int i=1;i<=1000;++i)
        if(vis[i]) add(id3[i],decc,i*i);
}
 
bool bfs()
{
    while(!q.empty()) q.pop();
    for(int i=0;i<=decc;++i) cur[i]=front[i],lev[i]=-1;
    int now;
    q.push(src);
    lev[src]=0;
    while(!q.empty())
    {
        now=q.front();
        q.pop();
        for(int i=front[now];i;i=nxt[i])
            if(cap[i]>0 && lev[to[i]]==-1)
            {
                lev[to[i]]=lev[now]+1;
                q.push(to[i]);
            }
    }
    return lev[decc]!=-1;
}
 
int dinic(int now,int flow)
{
    if(now==decc) return flow;
    int delta,rest=0;
    for(int &i=cur[now];i;i=nxt[i])
        if(cap[i]>0 && lev[to[i]]>lev[now])
        {
            delta=dinic(to[i],min(flow-rest,cap[i]));
            if(delta)
            {
                rest+=delta;
                cap[i]-=delta;
                cap[i^1]+=delta;
                if(rest==flow) break;
            }
        }
    if(rest!=flow) lev[now]=-1;
    return rest;
}
     
void solve()
{
    while(bfs()) 
        ans-=dinic(src,inf);
    printf("%d",ans);
}   
 
int main()
{
    init();
    build();
    solve();
    return 0;
}