CF 704 D. Captain America

CF 704 D. Captain America

题目链接

题目大意：给出\(n\)个点的坐标，你要将每个点染成红色或者蓝色。染一个红色要付出\(r\)的代价，染一个蓝色要付出\(b\)的代价。有\(m\)个限制，每个限制为“第\(i\)行（或者第\(i\)列）两种颜色的球数量差不能超过\(d\)”。

对于第\(i\)行，有\(x\)个点，数量差不能超过\(d\)，假设最终红球的数量为\(R\)，则：
\[ R-(x-R)\leq d\\ (x-R)-R\leq d\\ \]
得到：
\[ \lceil\frac{x-d}{2} \rceil\leq R\leq \lfloor\frac{x+d}{2} \rfloor \]
这就是个上下界网络流。将每一行，每一列分别建成一个点。源点向行点连边，列点向汇点连边。对于给定的交叉点上的行列之间连边，容量为\(1\)。先跑有源汇可行流，如果\(r<b\)，就跑最大可行流，否则跑最小可行流。

记得当前弧优化。

代码：

#include<bits/stdc++.h>
#define ll long long
#define N 200005

using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}

int n,m;
ll r,b;
int x[N],y[N];
int t[N],l[N],d[N];

struct road {
    int to,nxt;
    int f;
}s[N<<4];

int h[N<<1],cnt=1;
void add(int i,int j,int f) {
    if(!f) return ;
    s[++cnt]=(road) {j,h[i],f};h[i]=cnt;
    s[++cnt]=(road) {i,h[j],0};h[j]=cnt;
}

vector<int>px,py;
map<int,int>mpx,mpy;
int S,T;
int SS,TT;
ll ans;
int mnx[N],mny[N];
int totx[N],toty[N];
int nx,ny;
int ux[N],uy[N];
int dx[N],dy[N];
int dis[N];
queue<int>q;

bool dinic_bfs(int S,int T) {
    while(!q.empty()) q.pop();
    memset(dis,0x3f,sizeof(dis));
    q.push(S);
    dis[S]=0;
    while(!q.empty()) {
        int v=q.front();q.pop();
        for(int i=h[v];i;i=s[i].nxt) {
            int to=s[i].to;
            if(s[i].f&&dis[to]>dis[v]+1) {
                dis[to]=dis[v]+1;
                if(to==T) return 1;
                q.push(to);
            }
        }
    }
    return 0;
}

int dep;
bool ins[N];
int cur[N<<1];
int dfs(int v,int T,int maxf) {
    if(v==T) return maxf;
    ins[v]=1;
    int ret=0;
    for(int &i=h[v];i;i=s[i].nxt) {
        int to=s[i].to;
        if(s[i].f&&dis[to]==dis[v]+1&&!ins[to]) {
            int dlt=dfs(to,T,min(maxf,s[i].f));
            if(!dlt) dis[to]=0;
            s[i].f-=dlt;
            s[i^1].f+=dlt;
            ret+=dlt;
            maxf-=dlt;
            if(!maxf) break;
        }
    }
    ins[v]=0;
    return ret;
}

int dinic(int S,int T) {
    int ans=0;
    int dep=0;
    while(dinic_bfs(S,T)) {
        memcpy(cur,h,sizeof(h));
        ans+=dfs(S,T,1e9);
        memcpy(h,cur,sizeof(h));
    }
    return ans;
}

int e[N];
bool tag[N<<1];
void dfs(int v) {
    tag[v]=1;
    for(int i=h[v];i;i=s[i].nxt) {
        int to=s[i].to;
        if(s[i].f&&!tag[to]) dfs(to);
    }
}

int main() {
    n=Get(),m=Get();
    r=Get(),b=Get();
    for(int i=1;i<=n;i++) {
        x[i]=Get(),y[i]=Get();
        px.push_back(x[i]);
        py.push_back(y[i]);
        mpx[x[i]]=1;
        mpy[y[i]]=1;
    }
    
    sort(px.begin(),px.end());
    sort(py.begin(),py.end());
    px.resize(unique(px.begin(),px.end())-px.begin());
    py.resize(unique(py.begin(),py.end())-py.begin());
    for(int i=1;i<=n;i++) {
        x[i]=lower_bound(px.begin(),px.end(),x[i])-px.begin();
        y[i]=lower_bound(py.begin(),py.end(),y[i])-py.begin();
        totx[x[i]]++;
        toty[y[i]]++;
    }
    
    nx=px.size(),ny=py.size();
    for(int i=0;i<nx;i++) mnx[i]=totx[i];
    for(int i=0;i<ny;i++) mny[i]=toty[i];
    
    for(int i=1;i<=m;i++) {
        t[i]=Get(),l[i]=Get(),d[i]=Get();
        if(t[i]==1) {
            if(mpx.find(l[i])==mpx.end()) continue ;
            l[i]=lower_bound(px.begin(),px.end(),l[i])-px.begin();
            mnx[l[i]]=min(mnx[l[i]],d[i]);
        } else {
            if(mpy.find(l[i])==mpy.end()) continue ;
            l[i]=lower_bound(py.begin(),py.end(),l[i])-py.begin();
            mny[l[i]]=min(mny[l[i]],d[i]);
        }
    }
    
    S=nx+ny,T=nx+ny+1;
    SS=nx+ny+2,TT=nx+ny+3;
    int E=cnt+1;
    add(T,S,1e9);
    
    int tot1=0,tot2=0;
    for(int i=0;i<nx;i++) {
        dx[i]=totx[i]-mnx[i]+1>>1;
        ux[i]=totx[i]+mnx[i]>>1;
        tot1+=dx[i];
        add(S,i,ux[i]-dx[i]);
        if(dx[i]) add(SS,i,dx[i]);
        if(dx[i]>ux[i]) {
            cout<<-1;
            return 0;
        }
    }
    
    for(int i=0;i<ny;i++) {
        dy[i]=toty[i]-mny[i]+1>>1;
        uy[i]=toty[i]+mny[i]>>1;
        tot2+=dy[i];
        add(i+nx,T,uy[i]-dy[i]);
        if(dy[i]) add(i+nx,TT,dy[i]);
        if(dy[i]>uy[i]) {
            cout<<-1;
            return 0;
        }
    }
    
    for(int i=1;i<=n;i++) {
        e[i]=cnt+1;
        add(x[i],y[i]+nx,1);
    }
    
    add(S,TT,tot1),add(SS,T,tot2);
    int x=dinic(SS,TT);
    
    if(x!=tot1+tot2) {
        cout<<-1;
        return 0;
    }
    s[E].f=s[E^1].f=0;
    int x2;
    if(r>b) {
        x2=dinic(S,T);
    } else {
        dinic(T,S);
    }
    for(int i=1;i<=n;i++) {
        if(s[e[i]].f==0) {
            ans+=b;
        } else {
            ans+=r;
        }
    }
    cout<<ans<<"\n";
    for(int i=1;i<=n;i++) {
        if(s[e[i]].f==0) {
            cout<<"b";
        } else {
            cout<<"r";
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/hchhch233/p/11056902.html