QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#807634#9564. Hey, Have You Seen My Kangaroo?alexz1205WA 0ms16724kbC++204.0kb2024-12-10 09:38:372024-12-10 09:38:47

Judging History

你现在查看的是最新测评结果

  • [2024-12-10 09:38:47]
  • 评测
  • 测评结果:WA
  • 用时:0ms
  • 内存:16724kb
  • [2024-12-10 09:38:37]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;

const int NM = 2e5, K = 200;

typedef long long int lint;
typedef complex<int> vect;
typedef set<vect, decltype([](vect a, vect b){return (array<int,2>){a.real(), a.imag()} < (array<int, 2>){b.real(), b.imag()};})> setV;

int n, m, k;

template<typename T, size_t SIZE>
struct Multarr{
	size_t N, M;
	T data[SIZE];

	Multarr(){
		N = n;
		M = m;
	}

	Multarr(int N, int M){
		this->N = N;
		this->M = M;
		memset(data, 0, sizeof(data));
	}

	T* operator[](size_t i){
		return &data[i*M];
	}

	T& operator[](vect cur){
		return data[cur.real()*M + cur.imag()];
	}
};

char buff[K+5];
int c1[K], c2[K];
int change[K];
Multarr<char, 2*NM> grid;
Multarr<int, NM> req;
Multarr<vect, NM> nex;
Multarr<setV, NM> con;

void setUpArrs(){
	grid.N = n;
	grid.M = m;
	memset(grid.data, 0, sizeof(grid.data));

	req.N = n;
	req.M = m;
	memset(req.data, 0, sizeof(req.data));

	nex.N = n;
	nex.M = m;
}

map<char, vect> dir = {{'U', vect(-1, 0)}, {'D', vect(1, 0)}, {'L', vect(0, -1)}, {'R', vect(0, 1)}};

vect move(vect cur, vect mov){
	vect v = cur+mov;
	v = vect(min(max(v.real(), 0), n-1), min(max(v.imag(), 0), m-1));
	if (grid[v] != '0'){
		return v;
	}
	return cur;
}

int ans[NM+1];

int main(){
	scanf("%d %d %d", &n, &m, &k);
	scanf("%s", buff);
	setUpArrs();
	for (int x = 0; x < n; x ++){
		scanf("%s", grid[x]);
	}
	size_t sz = 0;
	for (int x = 0; x < n; x ++){
		for (int y = 0; y < m; y ++){
			if (grid[x][y] != '0'){
				sz ++;

				vect cur(x, y);
				for (int z = 0; z < k; z ++){
					cur = move(cur, dir[buff[z]]);
				}
				nex[x][y] = cur;
				con[cur].insert(vect(x, y));
				req[cur] ++;
			}
		}
	}
	vector<vect> rem;
	for (int x = 0; x < n; x ++){
		for (int y = 0; y < m; y ++){
			if (grid[x][y] != '0'){
				vect cur(x, y);
				if (!req[cur]) rem.push_back(cur);
			}
		}
	}

	memset(ans, -1, sizeof(ans));
	memset(ans+sz, 0, sizeof(ans) - sz*sizeof(int));

	int mov = 0;
	size_t i = sz-1;
	{
		setV inter;
		for (int x = 0; x < n; x ++){
			for (int y = 0; y < m; y ++){
				if (req[x][y] >= 2){
					inter.insert(vect(x, y));
				}
			}
		}
		memset(change, 0, sizeof(change));
		setV spect;
		for (vect v: inter){
			for (vect v2: con[v]){
				spect.insert(v2);
			}
		}
		for (int x = 0; x < k; x ++){
			setV newSpect;
			for (vect cur: spect){
				vect v = move(cur, dir[buff[x]]);
				newSpect.insert(v);
			}
			change[x] = spect.size() - newSpect.size();
			swap(spect, newSpect);
		}
	}
	while (!rem.empty()){
		for (int x = 0; x < k; x ++){
			sz -= change[x];
			if (sz < 0){
				exit(0);
			}
			while (i >= sz){
				ans[i--] = mov*k + x + 1;
			}
		}
		setV recalc;
		vector<vect> nexRem;
		for (vect cur: rem){
			if (req[nex[cur]] == 2){
				recalc.insert(nex[cur]);
			}
		}
		memset(c1, 0, sizeof(c1));
		memset(c2, 0, sizeof(c2));
		{
			setV spect;
			for (vect v: recalc){
				for (vect v2: con[v]){
					spect.insert(v2);
				}
			}
			for (int x = 0; x < k; x ++){
				setV newSpect;
				for (vect cur: spect){
					vect v = move(cur, dir[buff[x]]);
					newSpect.insert(v);
				}
				c1[x] = spect.size() - newSpect.size();
				swap(spect, newSpect);
			}
		}
		for (vect cur: rem){
			if (req[nex[cur]] == 2){
				recalc.erase(nex[cur]);
			}
			req[nex[cur]] --;
			con[nex[cur]].erase(cur);
			if (!req[nex[cur]]) nexRem.push_back(nex[cur]);
		}
		{
			setV spect;
			for (vect v: recalc){
				for (vect v2: con[v]){
					spect.insert(v2);
				}
			}
			for (int x = 0; x < k; x ++){
				setV newSpect;
				for (vect cur: spect){
					vect v = move(cur, dir[buff[x]]);
					newSpect.insert(v);
				}
				c2[x] = spect.size() - newSpect.size();
				swap(spect, newSpect);
			}
		}
		for (int x = 0; x < k; x ++){
			change[x] -= c1[x] - c2[x];
		}
		swap(nexRem, rem);
		mov ++;
	}


	for (int x = 1; x <= n*m; x ++){
		printf("%d\n", ans[x]);
	}
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 16712kb

input:

3 3 6
ULDDRR
010
111
010

output:

-1
4
2
1
0
0
0
0
0

result:

ok 9 numbers

Test #2:

score: -100
Wrong Answer
time: 0ms
memory: 16724kb

input:

3 3 6
ULDDRR
010
111
011

output:

-1
4
2
1
1
0
0
0
0

result:

wrong answer 1st numbers differ - expected: '7', found: '-1'