牛客国庆集训派对Day1.B.Attack on Titan(思路 最短路Dijkstra)

题目链接

---

(Description)

给定(n,m,C)及大小为((n+1)(m+1))的矩阵(c[i][j])。平面上有((n+1)(m+1))个点，从((0,0))编号到((n,m))。
在任意时刻，你可以选择在当前点((x,y))获取任意实数个单位的能量，获取每一单位需花费时间(c[x][y])；
也可以选择从一个点移动到另一个点，花费的能量是移动路线的欧几里得距离（只可以沿与坐标轴平行或与坐标轴夹角(45^{circ})的方向走），不花费时间。
初始时有(0)点能量，任意时刻的能量不能超过(C)（可以是实数）。求从((0,0))走到((n,m))所需的最少时间。
(n,mleq20, Cleq10000, c[i][j]leq10^6)。

(Solution)

最短路，但是怎么表示能量这一状态呢？
注意到每次走的路线长度，也就是能量花费，以及能量获取，一定可以表示成(xcdot1+ycdotsqrt2, x,yinmathbb{Z})（既存在花费也存在获取，所以(x,y)可能为正也可能为负）。
注意到图很小，容易发现(xin[-40,40],yin[-20,20])就够了，且(0leq xcdot1+ycdotsqrt2leq C)。
也就是任意时刻的能量只需要是(xcdot1+ycdotsqrt2)就可以了。把所有的合法(xcdot1+ycdotsqrt2)全算出值来，用这些做能量的状态就可以了，记为(w[i])。(C=10^4)的话状态数只有不到(1700)个。

然后我们就可以跑(Dijkstra)了？
对于状态((x,y,now))，表示当前在((x,y))，能量为(w[now])，只需要尝试向周围(8)个格子移动就好了，同时也可以选择在原地补充能量，由(w[now])变到到(w[now+1])。
还有个问题是(w[now])花费(1)或者(sqrt2)会变成(w)多少。这个预处理一下就好了，对每个(now)找到和(now-1)或(now-sqrt2)相差最近的（注意不要找错...= =）。

注意不要忘了(C)的限制，拿样例的状态数做数组大小= =（WA*INF）

---

//90ms	10320KB
#include <cmath>
#include <queue>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <algorithm>
#define eps 1e-7
#define eps2 1e-9
#define gc() getchar()
typedef long long LL;
const int N=23,M=1700,Way[5]={1,0,-1,0,1},Way2[5]={1,-1,-1,1,1};
const double sq2=sqrt(2),INF=(double)(1ll<<61);

int nxt1[M],nxt2[M];
bool vis[N][N][M];
double c[N][N],dis[N][N][M],w[M];
struct Node
{
	int x,y,now; double tm;
	bool operator <(const Node &x)const
	{
		return tm>x.tm;
	}
}e[M];
std::priority_queue<Node> q;

inline int read()
{
	int now=0;register char c=gc();
	for(;!isdigit(c);c=gc());
	for(;isdigit(c);now=now*10+c-48,c=gc());
	return now;
}

int main()
{
	int n=read(),m=read(),C=read(),cnt=0;
	for(int i=0; i<=n; ++i)
		for(int j=0; j<=m; ++j) c[i][j]=read();
	for(int i=-40; i<=40; ++i)
		for(int j=-20; j<=20; ++j)
		{
			double v=i+j*sq2;
			if(v>=0 && v<C+eps2) w[++cnt]=v;
		}
	std::sort(w+1,w+1+cnt);
	for(int i=1; i<=cnt; ++i)//走1单位或走sqrt(2)单位，能量会消耗到哪里 
	{
		for(int j=i-1; j; --j)
			if(fabs(w[i]-1-w[j])<eps) {nxt1[i]=j; break;}
//			if(w[i]-w[j]<1+eps) {nxt1[i]=j; break;}//wrong.
		for(int j=i-1; j; --j)
			if(fabs(w[i]-sq2-w[j])<eps) {nxt2[i]=j; break;}
	}
	for(int i=0; i<=n; ++i)
		for(int j=0; j<=m; ++j)
			for(int k=1; k<=cnt; ++k) dis[i][j][k]=INF;
	dis[0][0][1]=0, q.push((Node){0,0,1,0});
	while(!q.empty())
	{
		Node tmp=q.top(); q.pop();
		int x=tmp.x,y=tmp.y,now=tmp.now;
		if(vis[x][y][now]) continue;
		if(x==n&&y==m) return printf("%.9lf
",dis[x][y][now]),0;
		vis[x][y][now]=1;
		int t=0;
		if(w[now]+eps>1)
		{
			for(int i=0,xn,yn; i<4; ++i)
				if((xn=x+Way[i])<=n&&xn>=0&&(yn=y+Way[i+1])<=m&&yn>=0)
					e[++t]=(Node){xn,yn,nxt1[now],0};
		}
		if(w[now]+eps>sq2)
		{
			for(int i=0,xn,yn; i<4; ++i)
				if((xn=x+Way2[i])<=n&&xn>=0&&(yn=y+Way2[i+1])<=m&&yn>=0)
					e[++t]=(Node){xn,yn,nxt2[now],0};
		}
		if(now<cnt) e[++t]=(Node){x,y,now+1,(w[now+1]-w[now])*c[x][y]};
		#define v e[i]
		double ds=dis[x][y][now];
		for(int i=1; i<=t; ++i)
			if(dis[v.x][v.y][v.now]>ds+v.tm)
				v.tm=dis[v.x][v.y][v.now]=ds+v.tm, q.push(v);
	}

	return 0;
}