글 읽기 - RTE 가 납니다!

sgc109 7년 전 0

어디가 문제인지 찾지못하고있습니다.

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <queue>
#include <utility>
#include <stack>
#include <cstring>
#include <cstdlib>
#include <string>
#include <unordered_set>
#include <set>
#define REP(i,a,b) for(int i = a; i < b;++i) 
#define FOR(i,n) REP(i,0,n)
#define mp make_pair
#define pb push_back

using namespace std;

typedef long long ll;
typedef pair<int, int> ii;
typedef pair<ll, ll> llll;
typedef vector<int> vi;
typedef vector<ll> vll;
typedef vector<vector<int> > vvi;
typedef vector<ii> vii;
	
const int INF = 0x3a3a3a3a;
const int MAX_N = 1000;

const long long Hash = 10000007;

int h[Hash];

int dx[4] = {-1,0,0,1};
int dy[4] = {0,1,-1,0};

int map[MAX_N][MAX_N];
int visited[MAX_N][MAX_N];
int water;

int r,c, nwI, nwJ;
long long val;

vector<int> not0s;

bool inRange(int i, int j) {
	return (0 <= i && i < r && 0<= j && j < c);
}

void dfs(int hereI, int hereJ) {
	visited[hereI][hereJ] = 1;

	FOR(k,4) {
		int thereI = hereI + dy[k];
		int thereJ = hereJ + dx[k];
		if(inRange(thereI,thereJ) && !visited[thereI][thereJ] && map[thereI][thereJ] > water) {
			int a = thereI - nwI;
			int b = thereJ - nwJ;
			val = (val * 1000000 + (a*1000+b)) % Hash;
			dfs(thereI, thereJ);
		}
	}
}


int main() {
	scanf("%d%d",&r,&c);
	FOR(i,r) {
		FOR(j,c) {
			scanf("%d",&map[i][j]);
		}
	}
	int q;
	scanf("%d",&q);
	FOR(k,q) {
		not0s.clear();
		FOR(i,r) FOR(j,c) visited[i][j] = 0;
		FOR(i,Hash) h[i] = 0;
		scanf("%d",&water);
		FOR(i,r) {
			FOR(j,c) {
				if(!visited[i][j] && map[i][j] > water) {
					nwI = i; nwJ = j;
					val = 0;
					dfs(i,j);
					++h[val];
					if(h[val] == 1) not0s.pb(val);
				}
			}
		}
		long long ans = 0;
		FOR(i,not0s.size()) {
			long long cnt = h[not0s[i]];
			// printf("(%lld)",cnt);
			cnt = cnt * cnt - cnt;
			cnt /= 2;
			ans += cnt;
		}
		printf("%lld\n",ans);
	}
	return 0;
}

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

RTE 가 납니다!

sgc109 7년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말