洛谷P3749 [六省联考2017]寿司餐厅

传送门

题解

这几道都是上周llj讲的题，题解也写得十分好了，所以直接贴了几个链接和代码。

//Achen
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<vector>
#include<queue>
#include<cmath>
#include<ctime>
#include<map>
#define For(i,a,b) for(int i=(a);i<=(b);i++)
#define Rep(i,a,b) for(int i=(a);i>=(b);i--)
const int N=,inf=0x7fffffff;
typedef long long LL;
using namespace std;
int n,m,s,t,num,D[][],id[][],ans;
map<int,int>mp;

template<typename T> void read(T &x) {
    T f=; x=; char ch=getchar();
    while(ch!='-'&&(ch<''||ch>'')) ch=getchar();
    if(ch=='-') f=-,ch=getchar();
    for(;ch>=''&&ch<='';ch=getchar()) x=x*+ch-''; x*=f;
}

struct edge {
    int u,v,fl,cap,nx;
    edge(){}
    edge(int u,int v,int fl,int cap,int nx):u(u),v(v),fl(fl),cap(cap),nx(nx){}
}e[N]; 

int fir[N],cur[N],ecnt=,p[N];
void add(int u,int v,int cap) {
    e[++ecnt]=edge(u,v,,cap,fir[u]); fir[u]=ecnt;
    e[++ecnt]=edge(v,u,,,fir[v]); fir[v]=ecnt;
}

int calc(int s,int t) {
    int fl=inf;
    for(int i=t;i!=s;i=e[p[i]].u)
        fl=min(fl,e[p[i]].cap-e[p[i]].fl);
    for(int i=t;i!=s;i=e[p[i]].u)
        e[p[i]].fl+=fl,e[p[i]^].fl-=fl;
    return fl;
} 

int d[N],c[N];
queue<int>que;
void bfs(int s,int t) {
    for(int i=;i<=n;i++) cur[i]=fir[i],d[i]=n,c[i]=;
    que.push(t); d[t]=;
    while(!que.empty()) {
        int x=que.front();
        que.pop();
        for(int i=fir[x];i;i=e[i].nx) if(e[i].cap==) {
            int y=e[i].v;
            if(d[y]==n) {
                d[y]=d[x]+;
                que.push(y);
            }
        }
    }
}

int ISAP(int s,int t) {
    int res=;
    bfs(s,t);
    for(int i=;i<=n;i++) c[d[i]]++;
    for(int x=s;d[x]<n;) {
        if(x==t) {
            res+=calc(s,t);
            x=s;
        }
        int ok=;
        for(int &i=cur[x];i;i=e[i].nx)
            if(e[i].fl<e[i].cap&&d[e[i].v]+==d[x]) {
                p[x=e[i].v]=i; ok=; break;
            }
        if(!ok) {
            int M=n; cur[x]=fir[x];
            for(int i=fir[x];i;i=e[i].nx) if(e[i].cap>e[i].fl) {
                M=min(M,d[e[i].v]+);
            }
            if(!(--c[d[x]])) break;
            c[d[x]=M]++;
            if(x!=s) x=e[p[x]].u;
        }
    }
    return res;
}

int main() {
    read(n); read(m);
    s=++num; t=++num; num+=n;
    For(i,,n) {
        int col; read(col);
        if(!mp[col]) {
            mp[col]=++num;
            add(num,t,m*col*col);
        }
        add(i+,mp[col],inf);
        add(i+,t,col);
    }
    For(i,,n)
        For(j,i,n) {
            read(D[i][j]); id[i][j]=++num;
            if(D[i][j]>=) {
                add(s,num,D[i][j]);
                ans+=D[i][j];
            }
            else add(num,t,-D[i][j]);
            add(num,i+,inf);
            add(num,j+,inf);
        }
    For(i,,n)
        For(j,i,n) if(i!=j) {
            add(id[i][j],id[i+][j],inf);
            add(id[i][j],id[i][j-],inf);
        }
    n=num;
    ans-=ISAP(s,t);
    printf("%d\n",ans);
    return ;
}