poj2112

二分加网络流,开始虽然写了floyd,但是没有调用,导致错了许多次。

View Code
   
     
#include < iostream >
#include
< cstdio >
#include
< cstdlib >
#include
< cstring >
using namespace std;

#define N 300
#define inf 10000000

struct Edge
{
int x, y, nxt;
int c;
} bf[N
* N];

int k, c, m, n, s, t;
int map[N][N];
int ne, head[N], cur[N], ps[N * N], dep[N];

void floyed()
{
for ( int i = 1 ; i <= n; i ++ )
for ( int j = 1 ; j <= n; j ++ )
for ( int k = 1 ; k <= n; k ++ )
if (map[j][i] != inf && map[i][k] != inf)
if (map[j][i] + map[i][k] < map[j][k])
map[j][k]
= map[j][i] + map[i][k];
}

void addedge( int x, int y, int c)
{
bf[ne].x
= x;
bf[ne].y
= y;
bf[ne].c
= c;
bf[ne].nxt
= head[x];
head[x]
= ne ++ ;
bf[ne].x
= y;
bf[ne].y
= x;
bf[ne].c
= 0 ;
bf[ne].nxt
= head[y];
head[y]
= ne ++ ;
}

int flow( int n, int s, int t)
{
int tr, res = 0 ;
int i, j, k, f, r, top;
while ( 1 )
{
memset(dep,
- 1 , n * sizeof ( int ));
for (f = dep[ps[ 0 ] = s] = 0 , r = 1 ; f != r;)
for (i = ps[f ++ ], j = head[i]; j; j = bf[j].nxt)
{
if (bf[j].c && - 1 == dep[k = bf[j].y])
{
dep[k]
= dep[i] + 1 ;
ps[r
++ ] = k;
if (k == t)
{
f
= r;
break ;
}
}
}
if ( - 1 == dep[t])
break ;
memcpy(cur, head, n
* sizeof ( int ));
for (i = s, top = 0 ;;)
{
if ((i == t))
{
for (k = 0 , tr = inf; k < top; ++ k)
if (bf[ps[k]].c < tr)
tr
= bf[ps[f = k]].c;
for (k = 0 ; k < top; ++ k)
bf[ps[k]].c
-= tr, bf[ps[k] ^ 1 ].c += tr;
res
+= tr;
i
= bf[ps[top = f]].x;
}
for (j = cur[i]; cur[i]; j = cur[i] = bf[cur[i]].nxt)
if (bf[j].c && dep[i] + 1 == dep[bf[j].y])
break ;
if (cur[i])
{
ps[top
++ ] = cur[i];
i
= bf[cur[i]].y;
}
else
{
if ( 0 == top)
break ;
dep[i]
= - 1 ;
i
= bf[ps[ -- top]].x;
}
}
}
return res;
}

bool ok( int d)
{
ne
= 2 ;
memset(head,
0 , sizeof (head));
s
= 0 ;
t
= n + 1 ;
for ( int i = 1 ; i <= k; i ++ )
addedge(i, t, m);
for ( int i = k + 1 ; i <= n; i ++ )
addedge(s, i,
1 );
for ( int i = k + 1 ; i <= n; i ++ )
for ( int j = 1 ; j <= k; j ++ )
if (map[i][j] <= d)
addedge(i, j,
1 );
int ret = flow(n + 2 , s, t);
return (ret == c);
}

int binarysearch()
{
int l = 0 ;
int r = 20000 ;
int mid;
while (l < r)
{
mid
= (l + r) / 2 ;
if (ok(mid))
r
= mid;
else
l
= mid + 1 ;
}
return l;
}

int main()
{
// freopen("t.txt", "r", stdin);
scanf( " %d%d%d " , & k, & c, & m);
n
= c + k;
for ( int i = 1 ; i <= n; i ++ )
for ( int j = 1 ; j <= n; j ++ )
{
scanf(
" %d " , & map[i][j]);
if (map[i][j] == 0 && (i != j))
map[i][j]
= inf;
}
floyed();
int ans = binarysearch();
printf(
" %d\n " , ans);
return 0 ;
}

你可能感兴趣的:(poj)