HAOI2008 移动玩具

Posted 2021-12-02 fengxunling

题目链接：戳我

非常抱歉，菜鸡风浔凌又来水蓝题了qwq

看到题解上写双向搜索？？？什么鬼.......

看到数据范围特别小，直接划分二分图，两个点之间连接容量为1，费用为曼哈顿距离的边，跑费用流即可。

什么？你问我那个不能移动到有玩具的格子的限制？

不用管了啦，因为费用流会给你跑费用最小的，也就是每个点都会找到最近的那个点，不会有其他点和它抢位置的。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define S 0
#define T tot1+tot2+1
#define MAXN 110
using namespace std;
int n,m,tot1,tot2,t=1,c,f;
int pre_e[MAXN],pre_v[MAXN];
int a[MAXN][MAXN],b[MAXN][MAXN],head[MAXN],dis[MAXN],done[MAXN];
int move_x[4]={0,0,1,-1},move_y[4]={1,-1,0,0};
struct Node{int x,y,id;}node1[MAXN],node2[MAXN];
struct Edge{int nxt,to,dis,cost;}edge[100010];
inline void add(int from,int to,int dis,int cost)
{
    edge[++t].nxt=head[from],edge[t].to=to,edge[t].dis=dis,edge[t].cost=cost,head[from]=t;
    edge[++t].nxt=head[to],edge[t].to=from,edge[t].dis=0,edge[t].cost=-cost,head[to]=t;
}
inline bool spfa()
{
    queue<int>q;
    memset(dis,0x3f,sizeof(dis));
    memset(done,0,sizeof(done));
    q.push(S);done[S]=1;dis[S]=0;
    while(!q.empty())
    {
        int u=q.front();q.pop();done[u]=0;
        for(int i=head[u];i;i=edge[i].nxt)
        {
            int v=edge[i].to;
            if(edge[i].dis&&dis[u]+edge[i].cost<dis[v])
            {
                dis[v]=dis[u]+edge[i].cost;
                pre_e[v]=i,pre_v[v]=u;
                if(!done[v])
                    q.push(v),done[v]=1;
            }
        }
    }
    if(dis[T]>=0x3f3f3f3f) return false;
    int flow=0x3f3f3f3f;
    for(int i=T;i!=S;i=pre_v[i]) flow=min(flow,edge[pre_e[i]].dis);
    for(int i=T;i!=S;i=pre_v[i]) edge[pre_e[i]].dis-=flow,edge[pre_e[i]^1].dis+=flow;
    f+=flow;
    c+=flow*dis[T];
    return true;
}
int main()
{
    #ifndef ONLINE_JUDGE
    freopen("ce.in","r",stdin);
    #endif
    for(int i=1;i<=4;i++)
        for(int j=1;j<=4;j++)
        {
            scanf("%1d",&a[i][j]);
            if(a[i][j]==1) node1[++tot1]=(Node){i,j,tot1};
        }
    for(int i=1;i<=4;i++)
        for(int j=1;j<=4;j++)
        {
            scanf("%1d",&b[i][j]);
            if(b[i][j]==1) node2[++tot2]=(Node){i,j,tot2};
        }
    // printf("tot1=%d tot2=%d\n",tot1,tot2);
    for(int i=1;i<=tot1;i++) add(S,node1[i].id,1,0);
    for(int i=1;i<=tot2;i++) add(node2[i].id+tot1,T,1,0);
    for(int i=1;i<=tot1;i++)
    {
        for(int j=1;j<=tot2;j++)
        {
            add(node1[i].id,node2[j].id+tot1,1,abs(node1[i].x-node2[j].x)+abs(node1[i].y-node2[j].y));
        }
    }
    while(spfa());
    printf("%d\n",c);
    return 0;
}