CSP上的一道题,很巧的一道题,连续走小路的话,代价为连续小路的权值的和的平方,走大路的话,代价就直接为大路的权值。
简单想一下,就会发现小路和大路混杂在一起,很难考虑,如果简单地用bfs,记录节点时,就要记录当前节点已经走了多少小路。
而这就是我们要优化的地方,我们将所有小路都抽出来,跑一遍floyd,将所有小路的组合都拿出来,这样我们就可以限定小路不能连续走,然后bfs就只用记录当前节点是由小路来的还是大路来的。
用dist[N][2]记录,每次都拓展最近的节点,BFS就变成Dijkstra,当然SPFA也可以。
#include <bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define CLR(a) memset(a, 0, sizeof(a))
#define DBG(x) cout<<(#x)<<"="<<x<<endl
#define FOR(i, a, b) for(int i=(a); i<(b); i++)
#define REP(i, a, b) for(int i=(a); i<=(b); i++)
#define DOWN(i, a, b) for(int i=(a); i>=(b); i--)
using namespace std;
typedef