Hihocoder 第三十八周 二分答案

之前做算法题有做过这种用二分来寻找答案的，原来是一种算法。

可以找到符合答案的最值。

这个算法关键是确定时间复杂度，即能否使用。然后就是每个答案的判断函数应该怎么写。

题目：http://hihocoder.com/contest/hiho38/problem/1

原本我用的是深搜的，后来发现深搜有个弊端一直改不掉，vis数组标记的话又会造成超时。



比如这种，最大航行次数是2 的话，从1先访问 2 ， 再访问 3 ， 次数已经到了2 。但是2 ， 3 都会被标记。之后1 ，3路径就被忽略了。

如果用vis标记的话，回溯的时候又会超时，大家画个图就可以知道了。

最后还是没办法，逐层搜索就用bfs了，很想看看有没有人把dfs优化好了。

我的代码：

#include<iostream>
#include <cstring>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <queue>
using namespace std ;
bool ok  ;
struct Road{
int dest ;
int ww ;
Road(int d,int w){
dest = d , ww = w ;
}
};

vector<Road> v[20005];
int vis[20005] ;
int n , m , k ,t ;
/*   原来深搜的错误代码
void dfs(int x,const int& cost,int count)
{
if(x == t)
{
ok = true ;
return ;
}
int len = v[x].size();
for(int i = 0;i < len;i++)
if(!vis[v[x][i].dest] && v[x][i].ww <= cost && !ok && count + 1 <= k)
{
vis[v[x][i].dest] = 1 ;
dfs(v[x][i].dest , cost , count + 1) ;
}
}
*/
bool bfs(int cost)
{
queue<int> q ;
q.push(1) ;
vis[1] = 0 ;
while(!q.empty())
{
int u = q.front();
q.pop();
int len = v[u].size();
for(int i = 0;i < len;i++)
if(vis[v[u][i].dest]==-1 && v[u][i].ww <= cost && vis[u] + 1 <= k)
{
vis[v[u][i].dest] = vis[u] + 1 ;
q.push(v[u][i].dest);
}
}
return vis[t] != -1 && vis[t] <= k ;
}
bool check(int cost)
{
memset(vis , -1 , sizeof(vis));  // 原来是标记有没有找过的，现在作为计数
//ok = false ;
//dfs(1 , cost, 0 ) ;
//return ok ;
return bfs(cost) ;
}
int main()
{
while(~scanf("%d%d%d%d",&n,&m,&k,&t))
{
for(int i = 1;i <= n;i++)v[i].clear();

int u ,w, s ,min_ = 2147483647 , max_ = 0;
for(int i = 0;i < m;i++)
{
scanf("%d%d%d",&u,&s,&w);
v[u].push_back(Road(s, w));
v[s].push_back(Road(u , w));
max_ = max(max_ , w);
min_ = min(min_ , w);
}
int l = min_ , r = max_ ;
while(l < r)
{
int mid = (l + r) >> 1 ;
if(check(mid))r = mid;
else l = mid + 1 ;
}
printf("%d\n",l);

}
}