Cod sursa(job #2724168)

Utilizator FrostfireMagirescu Tudor Frostfire Data 16 martie 2021 17:24:32
Problema Car Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.76 kb
#include <iostream>
#include <fstream>
#include <deque>
#define NMAX 500
#define f first
#define s second
#define inf 2000000000

using namespace std;

ifstream fin("car.in");
ofstream fout("car.out");

int n, m, cost[NMAX+10][NMAX+10][8];
bool v[NMAX+10][NMAX+10], viz[NMAX+10][NMAX+10][8];
int dx[8] = {-1, -1, 0, 1, 1, 1, 0, -1};
int dy[8] = {0, 1, 1, 1, 0, -1, -1, -1};
deque <pair <pair <int, int>, int> > dq;
pair <int, int> p1, p2;

int main()
{
	fin >> n >> m >> p1.f >> p1.s >> p2.f >> p2.s;
	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			fin >> v[i][j];
	if(v[p1.f][p1.s] || v[p2.f][p2.s])
		{	fout << -1 << '\n';
			return 0;
		}
	for(int i=1; i<=n; i++)
		for(int j=1; j<=m; j++)
			for(int t=0; t<8; t++)
				cost[i][j][t] = inf;
	for(int t=0; t<8; t++)
		{	dq.push_back({p1, t});
			cost[p1.f][p1.s][t] = 0;
		}
	while(!dq.empty())
		{	pair <pair <int, int>, int> a = dq.front();
			dq.pop_front();
			int r = a.f.f, c = a.f.s, dir = a.s;
			if(viz[r][c][dir])
				continue;
			viz[r][c][dir] = 1;
			if(r == p2.f && c == p2.s)
				{	fout << cost[r][c][dir] << '\n';
					return 0;
				}
			int newdir = (dir + 1) % 8;
			if(1 + cost[r][c][dir] < cost[r][c][newdir])
				{	cost[r][c][newdir] = 1 + cost[r][c][dir];
					dq.push_back({{r, c}, newdir});
				}
			newdir = (dir + 7) % 8;
			if(1 + cost[r][c][dir] < cost[r][c][newdir])
				{	cost[r][c][newdir] = 1 + cost[r][c][dir];
					dq.push_back({{r, c}, newdir});
				}
			pair <int, int> vec = {dx[dir] + r, dy[dir] + c};
			if(vec.f && vec.s && vec.f <= n && vec.s <= m && !v[vec.f][vec.s] && cost[r][c][dir] < cost[vec.f][vec.s][dir])
				{	cost[vec.f][vec.s][dir] = cost[r][c][dir];
					dq.push_front({vec, dir});
				}
		}
	fout << -1 << '\n';
	return 0;
}