답안 #74421

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
74421 2018-09-01T06:37:48 Z cheetose 수중 미로 (FXCUP3_aqua) C++11
100 / 100
2389 ms 292164 KB
#include<cstdio>
#include<cstdlib>
#include<iostream>
#include<vector>
#include<queue>
#include<bitset>
#include<string>
#include<stack>
#include<set>
#include<unordered_set>
#include<map>
#include<unordered_map>
#include<cstring>
#include<complex>
#include<cmath>
#include<iomanip>
#include<numeric>
#include<algorithm>
#include<list>
#include<functional>
#include<cassert>
#define mp make_pair
#define pb push_back
#define X first
#define Y second
#define INF 1987654321987654321
#define PI 3.14159265358979323846264
#define fup(i,a,b,c) for(int (i)=(a);(i)<=(b);(i)+=(c))
#define fdn(i,a,b,c) for(int (i)=(a);(i)>=(b);(i)-=(c))
#define MEM0(a) memset(a,0,sizeof(a));
#define MEM_1(a) memset(a,-1,sizeof(a));
#define ALL(a) a.begin(),a.end()
#define SYNC ios_base::sync_with_stdio(false);cin.tie(0)
using namespace std;
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair<int, int> Pi;
typedef pair<ll, ll> Pll;
typedef pair<double, double> Pd;
typedef vector<int> Vi;
typedef vector<ll> Vll;
typedef vector<double> Vd;
typedef vector<Pi> VPi;
typedef vector<Pll> VPll;
typedef vector<Pd> VPd;
typedef tuple<int, int, int> iii;
typedef tuple<int, int, int, int> iiii;
typedef tuple<ll, ll, ll> LLL;
typedef vector<iii> Viii;
typedef vector<LLL> VLLL;
typedef complex<double> base;
const int MOD = 1000000009;
ll POW(ll a, ll b, ll MMM = MOD) { ll ret = 1; for (; b; b >>= 1, a = (a*a) % MMM)if (b & 1)ret = (ret*a) % MMM; return ret; }
ll gcd(ll a, ll b) { return b ? gcd(b, a%b) : a; }
ll lcm(ll a, ll b) { if (a == 0 || b == 0)return a + b; return a*(b / gcd(a, b)); }
int dx[] = { 0,1,0,-1 }, dy[] = { 1,0,-1,0 };
int ddx[] = { 0,0,1,1,1,-1,-1,-1 }, ddy[] = { 1,-1,1,0,-1,1,0,-1 };
 
char s[900][901];
VPll v[3240006];
ll dist[3240006];
bool chk[3240006];
int n;
inline int Z(int x,int y,int dir){return dir*n*n+x*n+y;}
const int SS=3240000,EE=3240001;
inline int Z2(int dir){return EE+dir+1;}
int main() {
	fill(dist,dist+3240006,INF);
	scanf("%d",&n);
	fup(i,0,n-1,1)scanf("%s",s[i]);
	Pi S,E;
	fup(i,0,n-1,1)
		fup(j,0,n-1,1)
	{
		if(s[i][j]=='M')
		{
			S=mp(i,j);
			if(i!=0&&s[i-1][j]!='W')v[Z2(3)].pb(mp(Z(i-1,j,3),0));
			if(i!=n-1&&s[i+1][j]!='W')v[Z2(2)].pb(mp(Z(i+1,j,2),0));
			if(j!=0&&s[i][j-1]!='W')v[Z2(1)].pb(mp(Z(i,j-1,1),0));
			if(j!=n-1&&s[i][j+1]!='W')v[Z2(0)].pb(mp(Z(i,j+1,0),0));
		}
		if(s[i][j]=='H')
		{
			E=mp(i,j);
			s[i][j]='.';
		}
	}
	fup(i,0,n-1,1)
		fup(j,0,n-1,1)
	{
		if(s[i][j]=='.')
		{
			if(i!=0&&s[i-1][j]=='.')v[Z(i,j,3)].pb(mp(Z(i-1,j,3),0));
			else if(i!=0&&s[i-1][j]=='M')v[Z(i,j,3)].pb(mp(Z2(3),0));
			
			if(i!=n-1&&s[i+1][j]=='.')v[Z(i,j,2)].pb(mp(Z(i+1,j,2),0));
			else if(i!=n-1&&s[i+1][j]=='M')v[Z(i,j,2)].pb(mp(Z2(2),0));
			
			if(j!=0&&s[i][j-1]=='.')v[Z(i,j,1)].pb(mp(Z(i,j-1,1),0));
			else if(j!=0&&s[i][j-1]=='M')v[Z(i,j,1)].pb(mp(Z2(1),0));
			
			if(j!=n-1&&s[i][j+1]=='.')v[Z(i,j,0)].pb(mp(Z(i,j+1,0),0));
			else if(j!=n-1&&s[i][j+1]=='M')v[Z(i,j,0)].pb(mp(Z2(0),0));
		}
	}
	fup(i,0,n-1,1)
	{
		int t=0, to=-1;
		fdn(j,n-1,0,1)
		{
			if(s[i][j]=='W')t++;
			else
			{
				if(to!=-1)
				{
					if(mp(i,j)==S)fup(kk,0,3,1)v[Z2(kk)].pb(mp(to,t*t));
					fup(kk,0,3,1)v[Z(i,j,kk)].pb(mp(to,t*t));
				}
				if(j!=0&&s[i][j-1]=='W')
				{
					to=Z(i,j,0);
					if(s[i][j]=='M')to=Z2(0);
					t=0;
				}
			}
		}
		t=0, to=-1;
		fup(j,0,n-1,1)
		{
			if(s[i][j]=='W')t++;
			else
			{
				if(to!=-1)
				{
					if(mp(i,j)==S)fup(kk,0,3,1)v[Z2(kk)].pb(mp(to,t*t));
					fup(kk,0,3,1)v[Z(i,j,kk)].pb(mp(to,t*t));
				}
				if(j!=n-1 && s[i][j+1]=='W')
				{
					to=Z(i,j,1);
					if(s[i][j]=='M')to=Z2(1);
					t=0;
				}
			}
		}
	}
	fup(j,0,n-1,1)
	{
		int t=0, to=-1;
		fdn(i,n-1,0,1)
		{
			if(s[i][j]=='W')t++;
			else
			{
				if(to!=-1)
				{
					if(mp(i,j)==S)fup(kk,0,3,1)v[Z2(kk)].pb(mp(to,t*t));
					fup(kk,0,3,1)v[Z(i,j,kk)].pb(mp(to,t*t));
				}
				if(i!=0&&s[i-1][j]=='W')
				{
					to=Z(i,j,2);
					if(s[i][j]=='M')to=Z2(2);
					t=0;
				}
			}
		}
		t=0, to=-1;
		fup(i,0,n-1,1)
		{
			if(s[i][j]=='W')t++;
			else
			{
				if(to!=-1)
				{
					if(mp(i,j)==S)fup(kk,0,3,1)v[Z2(kk)].pb(mp(to,t*t));
					fup(kk,0,3,1)v[Z(i,j,kk)].pb(mp(to,t*t));
				}
				if(i!=n-1&&s[i+1][j]=='W')
				{
					to=Z(i,j,3);
					if(s[i][j]=='M')to=Z2(3);
					t=0;
				}
			}
		}
	}
	priority_queue<Pll> q;
	fup(i,0,3,1)
	{
		v[SS].pb(mp(Z(S.X,S.Y,i),0));
		v[Z(E.X,E.Y,i)].pb(mp(EE,0));
	}
	dist[SS] = 0;
	q.push({ 0,SS });
	while (!q.empty())
	{
		ll x = q.top().Y;
		q.pop();
		if (chk[x])continue;
		chk[x] = 1;
		for (const auto& p : v[x])
		{
			ll next = p.X, cost = p.Y;
			if (dist[next] > dist[x]+cost)
			{
				dist[next] = dist[x] + cost;
				q.push({ -dist[next],next });
			}
		}
	}
	printf("%lld\n",dist[EE]==INF?-1:dist[EE]);
}

Compilation message

aqua.cpp: In function 'int main()':
aqua.cpp:70:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  scanf("%d",&n);
  ~~~~~^~~~~~~~~
aqua.cpp:71:21: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
  fup(i,0,n-1,1)scanf("%s",s[i]);
                ~~~~~^~~~~~~~~~~
# 결과 실행 시간 메모리 Grader output
1 Correct 94 ms 101752 KB Output is correct
2 Correct 83 ms 102004 KB Output is correct
3 Correct 83 ms 102080 KB Output is correct
4 Correct 85 ms 102668 KB Output is correct
5 Correct 85 ms 102968 KB Output is correct
6 Correct 93 ms 103712 KB Output is correct
7 Correct 97 ms 104280 KB Output is correct
8 Correct 97 ms 104364 KB Output is correct
9 Correct 93 ms 104364 KB Output is correct
10 Correct 82 ms 104364 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 94 ms 101752 KB Output is correct
2 Correct 83 ms 102004 KB Output is correct
3 Correct 83 ms 102080 KB Output is correct
4 Correct 85 ms 102668 KB Output is correct
5 Correct 85 ms 102968 KB Output is correct
6 Correct 93 ms 103712 KB Output is correct
7 Correct 97 ms 104280 KB Output is correct
8 Correct 97 ms 104364 KB Output is correct
9 Correct 93 ms 104364 KB Output is correct
10 Correct 82 ms 104364 KB Output is correct
11 Correct 336 ms 128080 KB Output is correct
12 Correct 552 ms 159824 KB Output is correct
13 Correct 1597 ms 234820 KB Output is correct
14 Correct 1251 ms 234820 KB Output is correct
15 Correct 313 ms 234820 KB Output is correct
16 Correct 2291 ms 292164 KB Output is correct
17 Correct 2389 ms 292164 KB Output is correct
18 Correct 1407 ms 292164 KB Output is correct
19 Correct 1844 ms 292164 KB Output is correct
20 Correct 122 ms 292164 KB Output is correct