1305: [CQOI2009]dance跳舞

题目链接

题目大意：给定n个男生和n个女生，一些互相喜欢而一些不，举行几次舞会，每次舞会要配成n对，不能有相同的组合出现，每个人只能与不喜欢的人跳k次舞，求最多举行几次舞会

题解：将一个人x拆成两个点，分别为lx和dx，表示喜欢/不喜欢

对男孩

连(lx,dx,k),表示消耗了一次跟不喜欢的人的跳舞机会
连(st,lx)，容量待定

对女孩

连(dx,lx,k),同上
连(lx,ed)，容量待定

对于两个人x,y

若x喜欢y， 连(lx,ly,1)
否则， 连(dx,dy,1)

当然可以二分，但是这里用了动态加点

当答案增大1，就给连接源/汇的边扩容1，直至无法满流，输出ans-1即可

我的收获：网络流枚举答案强啊

#include 
#include 
#include 
#include 
#include 
using namespace std;

const int M=234577;
#define T(x) (x)*n*m
#define INF 0x7fffffff
#define pii pair
const int dx[]={0,1,-1,0,0};
const int dy[]={0,0,0,1,-1};

int n,m,t,now,tim,cnt,tot,st,ed,door;
int id[25][25],dis[25][25];
int num[M],d[M],head[M],last[M];
bool Exit;
char s[25][25];

queue q;

struct edge{int to,c,nex;}e[M*10];

void add(int u,int v,int w){e[t]=(edge){v,w,head[u]};last[u]=head[u]=t++;}

void insert(int x,int y,int z){add(x,y,z),add(y,x,0);};

int dfs(int x,int in)
{
    if(x==ed) return in;
    int ans=0,f;
    for(int i=last[x];i!=-1;last[x]=i=e[i].nex)
    {
        int v=e[i].to;
        if(e[i].c&&d[v]==d[x]-1){
            f=dfs(v,min(in-ans,e[i].c));
            ans+=f;e[i].c-=f,e[i^1].c+=f;
            if(Exit||ans==in) return ans;
        }
    }
    if(--num[d[x]]==0) Exit=1;
    d[x]++,num[d[x]]++,last[x]=head[x];
    return ans;
}

int isap(int tim)
{
    memset(d,0,sizeof(d));
    Exit=0;
    int flow=0;
    while(!Exit) flow+=dfs(st,INF);
    return flow;
}

void build()
{
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            if(s[i][j]=='D')
            {
                if(tim) insert(id[i][j]+T(tim-1),id[i][j]+T(tim),INF);
                insert(id[i][j]+T(tim),ed,1);
            }
}

void getdis(int x,int y)
{
    memset(dis,0,sizeof(dis));
    q.push(make_pair(x,y));
    while(!q.empty()){
        int nx=q.front().first,ny=q.front().second;q.pop();
        for(int k=1;k<=4;k++){
            int fx=nx+dx[k],fy=ny+dy[k];
            if(dis[fx][fy]||fx<1||fx>n||fy<1||fy>m||s[fx][fy]!='.') continue;
            dis[fx][fy]=dis[nx][ny]+1;
            insert(id[fx][fy],id[x][y]+T(dis[fx][fy]-1),1);
            q.push(make_pair(fx,fy));
        }
    } 
}

void work()
{
    while(nowif(tim>=n*m){puts("impossible");return ;}
        build();
        tim++;now+=isap(tim);
    }
    cout<void init()
{
    memset(head,-1,sizeof(head));
    memset(last,-1,sizeof(last));
    cin>>n>>m;
    st=0,ed=T(n*m)+1;
    for(int i=1;i<=n;i++) scanf("%s",s[i]+1); 
    for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) id[i][j]=++cnt;//！！！！ 
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++){
            if(s[i][j]=='.') tot++,insert(st,id[i][j],1);
            if(s[i][j]=='D') getdis(i,j);
        }
}

int main()
{
    init();
    work();
    return 0;
}