标签：return int 题解 cin Wireless ++ computer POJ find

POJ 2236 Wireless Network

加工并储存数据的数据结构

并查集

这是并查集的基本应用，两台修好的电脑若距离d内则加入合并。不过不小心的话会TLE，比如：

#include <iostream>
using namespace std;
 
#define MAX_N 1001 + 16
int parent[MAX_N];
int height[MAX_N];
bool status[MAX_N];
int distance[MAX_N][MAX_N];
 
void init(const int& n)
{
	for (int i = 0; i < n; ++i)
	{
		parent[i] = i;
		height[i] = 0;
	}
}
 
int find(const int& x)
{
	if (parent[x] == x)
	{
		return x;
	}
	else
	{
		return parent[x] = find(parent[x]);
	}
}
 
void unite(int x, int y)
{
	x = find(x);
	y = find(y);
	if (x == y)
	{
		return;
	}
 
	if (height[x] < height[y])
	{
		parent[x] = y;
	}
	else
	{
		parent[y] = x;
		if (height[x] == height[y])
		{
			++height[x];
		}
	}
}
 
bool same(const int& x, const int& y)
{
	return find(x) == find(y);
}
 
pair<int, int> computer[MAX_N];
int square(const int& x)
{
	return x * x;
}

int main(int argc, char *argv[])
{
	int N, d;
	cin >> N >> d;
	for (int i = 0; i < N; ++i)
	{
		cin >> computer[i].first >> computer[i].second;
	}
	init(N);
	char operation;
	int x, y;
	while (cin >> operation)
	{
		if (operation == 'O')
		{
			cin >> x;
			--x;
			status[x] = true;
			for (int i = 0; i < N; ++i)
			{
				if (i == x)
				{
					continue;
				}
				if (status[i] && square(computer[x].first - computer[i].first) + square(computer[x].second - computer[i].second) <= square(d))
				{
					unite(x, i);
				}
			}
		}
		else
		{
			cin >> x >> y;
			--x; --y;
			if (same(x, y))
			{
				cout << "SUCCESS" << endl;
			}
			else
			{
				cout << "FAIL" << endl;
			}
		}
	}
	return 0;
}

平方计算太多了，初始化的时候算一次记录在一个二维数组中就够了。

#include<iostream>
#include<cmath>
#include<cstring>
using namespace std;

#define ms(a,b) memset(a,b,sizeof(a));
#define maxn  1010

int f[maxn];
int h[maxn];
bool status[maxn];//电脑是否维修好了
bool able[maxn][maxn];//distance

void init(int n) {
    //电脑编号从0开始
    for (int i = 0; i < n; ++i) {
        f[i] = i;
        h[i] = 0;
    }
}

int find(int x) {
    return f[x] == x ? x : f[x] = find(f[x]);
}

void merge(int x, int y) {
    x = find(x);
    y = find(y);
    if (x == y)return;
    if (h[x] < h[y])f[x] = y;
    else {
        f[y] = x;
        if (h[x] == h[y])++h[x];
    }
}

bool same(int x, int y) {
    return find(x) == find(y);
}

pair<int, int>computer[maxn];
int square(int x) {
    return x * x;
}

int main() {
    int N, d;
    cin >> N >> d;
    for (int i = 0; i < N; ++i)
    {
        cin >> computer[i].first >> computer[i].second;
    }
    init(N);
    for (int i = 0; i < N; ++i)
    {
        for (int x = i; x < N; ++x)
        {
            if (square(computer[x].first - computer[i].first) + square(computer[x].second - computer[i].second) <= square(d))
            {
                able[i][x] = true;
                able[x][i] = true;
            }
        }
    }
    char operation;
    int x, y;
    while (cin >> operation) {
        if (operation == 'O')
        {
            cin >> x;
            --x;
            status[x] = true;
            for (int i = 0; i < N; ++i)
            {
                if (i == x)
                {
                    continue;
                }
                if (status[i] && able[x][i])
                {
                    merge(x, i);
                }
            }
        }
        else
        {
            cin >> x >> y;
            --x; --y;
            if (same(x, y))
            {
                cout << "SUCCESS" << endl;
            }
            else
            {
                cout << "FAIL" << endl;
            }
        }
    }
}

标签：return,int,题解,cin,Wireless,++,computer,POJ,find
来源： https://www.cnblogs.com/RioTian/p/12864935.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。