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());
}