vijos 1521 跳舞 【二分】【最大流】

描述

一次舞会有n个男孩和n个女孩。每首曲子开始时，所有男孩和女孩恰好配成n对跳交谊舞。每个男孩都不会和同一个女孩跳两首（或更多）舞曲。

有一些男孩女孩相互喜欢，而其他相互不喜欢（不会“单向喜欢”）。每个男孩最多只愿意和k个不喜欢的女孩跳舞，而每个女孩也最多只愿意和k个不喜欢的男孩跳舞。

给出每对男孩女孩是否相互喜欢的信息，舞会最多能有几首舞曲？

格式

输入格式

第一行包含两个整数n和k。以下n行每行包含n个字符，其中第i行第j个字符为'Y'当且仅当男孩i和女孩j相互喜欢。

输出格式

仅一个数，即舞曲数目的最大值。

样例1

样例输入1

3 0
YYY
YYY
YYY

 
 

  
  

   
   Copy
  
  
 
 

样例输出1

3

 
 

  
  

   
   Copy
  
  
 
 

样例2

样例输入2

3 0
YYY
YYN
YNY

 
 

  
  

   
   Copy
  
  
 
 

样例输出2

2

 
 

  
  

   
   Copy
  
  
 
 

样例3

样例输入3

2 0
YN
YN

 
 

  
  

   
   Copy
  
  
 
 

样例输出3

0

 
 

  
  

   
   Copy
  
  
 
 

样例4

样例输入4

2 1
YN
YN

 
 

  
  

   
   Copy
  
  
 
 

样例输出4

1

点击打开链接

#include<bits/stdc++.h>
using namespace std;
#define mms(x, y) memset(x, y, sizeof x)
const int MAX = 205;
const int INF = 0x3f3f3f3f;
struct node
{
    int x, y, l, next;
} a[MAX * MAX];
int n, k, m[MAX][MAX], l, r, pre[MAX], num, T, lev[MAX];
void add(int x, int y, int z)
{
    a[++num].x = x;
    a[num].y = y;
    a[num].l = z;
    a[num].next = pre[x];
    pre[x] = num;
    a[++num].x = y;
    a[num].y = x;
    a[num].l = 0;
    a[num].next = pre[y];
    pre[y] = num;
}
bool bfs()
{
    queue <int> q;
    mms(lev, 0);
    lev[0] = 1;
    q.push(0);
    while(!q.empty())
    {
        int v = q.front();
        q.pop();
        for(int i = pre[v]; i; i = a[i].next)
        {
            int u = a[i].y;
            if(a[i].l && !lev[u])
            {
                lev[u] = lev[v] + 1;
                if(u == T)
                    return 1;
                q.push(u);
            }
        }
    }
    return 0;
}
int dfs(int x, int s)
{
    if(x == T || !s)
        return s;
    int S = s;
    for(int i = pre[x]; i && s; i = a[i].next)
    {
        int u = a[i].y;
        if(lev[u] == lev[x] + 1)
        {
            int t = dfs(u, min(a[i].l, s));
            s -= t;
            a[i].l -= t;
            a[i ^ 1].l += t;
        }
    }
    if(S == s)
        lev[x] = -1;
    return S - s;
}
bool getans(int x)
{
    num = 1;
    mms(pre, 0);
    T = n * 4 + 1;
    for(int i = 1; i <= n; i++)
    {
        add(0, i, x);
        add(i, n + i, k);
        add(n * 3 + i, n * 2 + i, k);
        add(n * 2 + i, T, x);
    }
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= n; j++)
            if(m[i][j])
                add(i, n * 2 + j, 1);
            else
                add(n + i, n * 3 + j, 1);
    int ans = 0;
    while(bfs())
        ans += dfs(0, INF);
    return ans == n * x ? 1 : 0;
}
void Bsearch()
{
    l = 0;
    r = n;
    while(l < r)
    {
        int mid = (l + r) >> 1;
        if(l == r - 1)
            mid++;
        if(getans(mid))
            l = mid;
        else
            r = mid - 1;
    }
}
int main()
{
    char s[MAX];
    scanf("%d%d", &n, &k);
    for(int i = 1; i <= n; i++)
    {
        scanf("%s", s + 1);
        for(int j = 1; j <= n; j++)
            m[i][j] = s[j] == 'Y' ? 1 : 0;
    }
    Bsearch();
    printf("%d\n", l);
    return 0;
}