nyoj 130 相同的雪花

存储雪花时可以把雪花的长度和一并存储起来,然后根据雪花的长度和做一个hash表。

判断2个雪花是否相同,就需要从雪花a和雪花b中找到雪花臂长度相同的2个位置。然后从这2个位置开始正序,逆序比较看是否能得到相同的序列。如果得不到,就从下一次长度相同的2个位置开始正序,倒序比较。

#include <cstdio>
#include <vector>
using namespace std;
#define max_n 10003
struct node {
	int snow[6];
	int sum;
};
bool cmp(int a[],int b[]);
int main()
{
	int l, n;
	scanf("%d",&l);
	while (l--)
	{
		node nw, cur;
		int i;
		vector<node> np[max_n];
		int ok = 0;
		scanf("%d",&n);
		while (n--)
		{
			int nsum = 0;
			for (i = 0; i < 6; i++)
			{
				scanf("%d",&nw.snow[i]);
				nsum += nw.snow[i];
			}
			nw.sum = nsum;
			nsum %= max_n;
			if (!ok)
			{
				for (i = 0; i != np[nsum].size(); i++)
				{
					if (nw.sum != np[nsum][i].sum)
					{
						continue;
					}
					cur = np[nsum][i];
					ok = cmp(nw.snow,cur.snow);
					if (ok)
					{
						break;
					}
				}
				if (!ok)
				{
					np[nsum].push_back(nw);
				}
			}
		}
		if (ok)
		{
			printf("Twin snowflakes found.\n");
		}
		else
		{
			printf("No two snowflakes are alike.\n");
		}
	}
	return 0;
}
bool cmp(int a[], int b[])
{
	int start = -1;
	int ok = 0;
	while (start < 6 && ok == 0)
	{
		for (start += 1;; start++)
		{
			if (start >= 6)
				return 0;
			if (a[start] == b[0])
				break;
		}
		ok = 1;
		for (int i = 0, j = start; i < 6; i++)
		{
			if (b[i] != a[j % 6])
			{
				ok = 0;
				break;
			}
		}
		if (!ok)
		{
			ok = 1;
			for (int i = 0, j = start + 6; i < 6; i++, j--)
			{
				if (b[i] != a[j % 6])
				{
					ok = 0;
					break;
				}
			}
		}
	}
	return ok;
}


你可能感兴趣的:(hash表,nyoj)