CF1196F

CF1196F

• 题目

  CF1196F

• 分析

  看到题懵逼了。。。

  赛后补题，真的发现数据范围很重要啊。。。

  题目要求任意起点终点的最短路径中的第 $k$ 短。最短路 $O(nlog(n))$ ，枚举起点 $O(n)$ ,总的复杂度 $O(n^{2}log(n))$ 显然通过不了。。这时看到 $k$ 的范围 $k\le min(\frac{n\times (n-1)}{2},400)$ ,只要算到第 $400$ 小，这样就可以在读入的时候枚举前 $400$ 小的边，用 $floyd$ 跑最短路。这样就必定能在新建的图中找到第 $k$ 短路。

  $n>400$ 是因为 $n-1\le m$ .

• 代码

  const int N = 4005;
  const int M = 2e5 + 5;
  ll d[N][N];
  struct node
  {
  	int u;
  	int v;
  	ll w;
  } a[M];
  bool cmp(node a, node b)
  {
  	return a.w < b.w;
  }
  int main ()
  {
  	//freopen("input.in", "r", stdin);
  	//freopen("test.out", "w", stdout);
  	int n, m, k;
  	read(n);
  	read(m);
  	read(k);
  	for (int i = 1; i <= 1001; i++)
  		for (int j = 1; j <= 1001; j++)
  			d[i][j] = (i == j ? 0 : (ll)1e15);
  	int cnt = 0 ;
  	if (n > 400) {
  		std::vector<node> vx;
  		for (int i = 1; i <= m; i++)
  		{
  			read(a[i].u);
  			read(a[i].v);
  			read(a[i].w);
  			vx.push_back((node) {a[i].u, a[i].v, a[i].w});
  		}
  		sort(vx.begin(), vx.end(), cmp);
  		map<int, int> mp;
  		while ((int)vx.size() > k)
  			vx.pop_back();
  		for (int i = 0; i < vx.size(); i++)
  		{
  			if (!mp[vx[i].u])
  				mp[vx[i].u] = ++cnt;
  			if (!mp[vx[i].v])
  				mp[vx[i].v] = ++cnt;
  		}
  		for (int i = 0; i < vx.size(); i++)
  		{
  			int u = mp[vx[i].u];
  			int v = mp[vx[i].v];
  			d[u][v] = d[v][u] = vx[i].w;
  		}
  		n = cnt;
  	}
  	else
  	{
  		for (int i = 1; i <= m; i++)
  		{
  			read(a[i].u);
  			read(a[i].v);
  			read(a[i].w);
  			d[a[i].u][a[i].v] = d[a[i].v][a[i].u] = a[i].w;
  		}
  	}
  	for (int k = 1; k <= n; k++)
  	{
  		for (int i = 1; i <= n; i++)
  			for (int j = 1; j <= n; j++)
  				d[i][j] = min(d[i][j], d[i][k] + d[k][j]);
  	}
  	std::vector<ll> v;
  	priority_queue<ll, vector<ll>, greater<ll> > q;
  	for (int i = 1; i <= n; i++)
  	{
  		for (int j = i + 1; j <= n; j++) {
  			q.push(d[i][j]);
  		}
  	}
  	for (int i = 1; i < k; i++) {
  		q.top();
  		q.pop();
  	}
  	cout << q.top() << endl;
  	return 0 ;
  }
  
  

• 题型

  $floyd$

​ 感悟：有些题目数据范围真的很重要，范围决定复杂度啊。。。