2019ICPC南京J - Spy

2019ICPC南京J - Spy

有两个队伍进行比拼
两边各有n个人
第一行表示Amy队伍每个人的战斗力
第二行表示击败每个人能获得的声望值
第三行表示Bob第一只队伍的战斗力
第三行表示Bob第二支队伍的战斗力
Bob的两支队伍的队员两两配，战斗力相加，然后和Amy的队员进行战斗。
每个人只能战斗一次。
如果Bob方的战斗力大于Amy的就获得声望值，如果战斗力≤Amy队员战斗力，就无法获得声望值。
双方的战斗匹配都是随机的，问Bob能获得的声望的期望是多少。
最后的答案乘以n并且保证是整数

Bob的队员组合 能战胜的 Amy的队员的声望值的期望乘以n 是 能战胜的Amy的队员的声望的总和。
所以转换成二分图最大权匹配，左右两边为Bob的队员，边的值为战胜的Amy的队员的声望的总和。
然后题目卡了dfs版本的KM。。。
嗯，只要换成bfs版本的KM就能过了orz

#include<bits/stdc++.h>
using namespace std;
#define ll long long

const int MAXN = 4e2+10;
const ll inf = 9e18;

struct node{
    node(ll _v = 0,ll _r = 0):v(_v),r(_r){}
    ll v,r;
}a[MAXN];

ll w[MAXN][MAXN];

bool cmp(const node &x,const node &y) {return x.v < y.v;}

int lx[MAXN],ly[MAXN],linky[MAXN],n,vis[MAXN],pre[MAXN];
ll slack[MAXN];

void bfs(int k)
{
    ll px,py = 0,yy = 0,d;
    memset(pre,0,sizeof(pre));
    for(int i=1;i<=n;i++)	slack[i]=inf;
    linky[py] = k;
    do
    {
        px = linky[py],d = inf,vis[py] = 1;
        for (int i = 1;i<=n;i++)
        {
            if (vis[i] == 0)
            {
                if (slack[i] >lx[px] + ly[i] - w[px][i])
                    slack[i] = lx[px] + ly[i] - w[px][i],pre[i] = py;
                if (slack[i] < d) d= slack[i],yy = i;
            }
        }
        for (int i = 0;i<=n;i++)
            if (vis[i]) lx[linky[i]] -= d,ly[i] += d;
            else slack[i] -= d;
        py = yy;
    } while (linky[py]);
    while (py) linky[py] = linky[pre[py]],py = pre[py];
}

ll km()
{
    memset(lx,0,sizeof(lx));
    memset(ly,0,sizeof(ly));
    for (int i = 1;i<=n;i++)
    {
        memset(vis,0,sizeof(vis));
        bfs(i);
    }
    ll res = 0;
    for (int i = 1;i<=n;i++)
        if (linky[i]) res += w[linky[i]][i];
    return res;
}

int main()
{
    ll b[MAXN],c[MAXN];
    scanf("%d",&n);
    for (int i = 1;i<=n;i++) scanf("%lld",&a[i].v);
    for (int i = 1;i<=n;i++) scanf("%lld",&a[i].r);
    sort(a+1,a+n+1,cmp);
    ll sum[MAXN] = {};
    for (int i = 1;i<=n;i++) sum[i] = sum[i-1] + a[i].r;
    for (int i = 1;i<=n;i++) scanf("%lld",&b[i]);
    for (int i = 1;i<=n;i++) scanf("%lld",&c[i]);
    for (int i = 1;i<=n;i++)
    {
        for (int j = 1;j<=n;j++)
        {
            node tmp(b[i] + c[j],0);
            int pos = lower_bound(a + 1,a+n+1,tmp,cmp) - a;
            w[i][j] = sum[pos-1];
        }
    }
    printf("%lld\n",km());
}