uva297 Quadtrees (树的重建)

题意:一幅图有1024个点, 可以对图平均分成4块, 并且子图也可以再往下分, 直到一个子图表示一个点。 f表示这块子图填满, p表示它还有4个子图, e表示没有子图(当然啦, 它也没有填满)。 给定两个字符串(其实就是两幅图, 两棵树), 求把两图合并后的图的黑点数。


思路:可以根据两字符串建两棵树, 然后合并。 但是直接用字符串合并似乎会更简单。从头到尾逐个字符比较两个字符串, 如果是ff、ee、pp则输出f、e、p到buff,如果是pe则输出p及p后面的4个字母(它的子图), 如果是pf则输出f, p后面4个字母(p当然也要读掉, 有比较的都要在原有字符串里读掉), 如果是pf, 则输出f。 合并好字符串后, 用递归的方式统计黑点数, 遇到p往下递归一层, 每往下递归一层f所代表的黑点数要除以4.


算法复杂度:o(N), N是最长的的那个字符串的长度。


代码:

#include <cstdio>
#include <cstring>
using namespace std;

#define MAX_LEN 3000
#define SUM_BIT 1024
#define PP 224
#define	FF 204
#define EE 202
#define PE 213
#define PF 214
#define EF 203

void doSame(char *&, char *&, char *&);
void doPE(char *&, char *&, char *&);
void doPF(char *&, char *&);
void countBit(char *&, int);

int count;
const int bit[6] = {1024, 256, 64, 16, 4, 1};

int main()
{
	int cases;
	scanf("%d%*c", &cases);
	while (cases--) {
		// init
		char str1[MAX_LEN];
		char str2[MAX_LEN];
		char rslt[MAX_LEN];
		int levelMrak[MAX_LEN];
		char *p1 = str1;
		char *p2 = str2;
		char *p = rslt;
		memset(str1, 0, sizeof(str1));
		memset(str2, 0, sizeof(str2));
		memset(rslt, 0, sizeof(rslt));
		memset(levelMrak, 0, sizeof(levelMrak));

		// enter
		gets(str1);
		gets(str2);

		// star
		while (true) {
			int sum = (int) (*p1 + *p2);
			if (sum == PP || sum == FF || sum == EE) {
				doSame(p1, p2, p);
			} else if (sum == PE) {
				if (*p1 == 'p') {
					doPE(p1, p2, p);
				} else {
					doPE(p2, p1, p);
				}
			} else if (sum == PF) {
				if (*p1 == 'f') {
					doPF(p1, p2);
				} else {
					doPF(p2, p1);
				}
				*p++ = 'f';
			} else if (sum == EF) {
				*p++ = 'f';
				p1++;
				p2++;
			}

			if (!*p1) {
				while (*p2) {
					*p++ = *p2++;
				}
				break;
			} else if (!*p2) {
				while (*p1) {
					*p++ = *p1++;
				}
				break;
			}
		}

		// count
		count = 0;
		if (rslt[0] == 'f') {
			count = SUM_BIT;
		} else if (rslt[0] == 'e') {
			count = 0;
		} else {
			p = rslt;
			p++;
			countBit(p, 1);	
		}

		// output
		printf("There are %d black pixels.\n", count);

	}

	return 0;
}

void doSame(char *&p1, char *&p2, char *&p)
{
	*p++ = *p1;
	p1++;
	p2++;
}

void doPE(char *&P, char *&E, char *&rslt)
{
	for (int i = 0; i < 5; i++) {
		*rslt++ = *P++;
	}
	E++;
}

void doPF(char *&F, char *&P)
{
	F++;
	for (int i = 0; i < 5; i++) {
		P++;
	}
}
void countBit(char *&p, int level)
{
	for (int i = 0; i < 4; i++) {
		if (*p == 'p') {
			p++;
			countBit(p, level + 1);
		} else if (*p == 'f') {
			count += bit[level];	
			p++;
		} else {
			p++;
		}
	}
}


阅读更多
个人分类: 基础数据结构
上一篇uva112 Tree Summing(二叉树 + 栈)
下一篇uva712 S-Tree (假二叉树重建)
想对作者说点什么? 我来说一句
相关热词

没有更多推荐了,返回首页

关闭
关闭
关闭