一个菜菜的蜕变之路?

时间是个常数,也是个变数

Bzoj P4195 [Noi2015]程序自动分析___并查集+hash

题目大意:

NX,Y,ZZ=1X=Y,Z=0XYYESNO

多组数据
1N106
1X,Y108
0Z1

分析:

显然可以用一个并查集维护,
我们将相等的点存在同一个集合中,当一个XYXY同时在一个集合中,则有冲突
因为XY比较大,可以采用hash或者map

代码:

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#define modn 1000007
#define N 1000005

using namespace std;

struct Node {
    int x, y, z;
}a[N];
int Hash[modn][2], f[modn], cnt;

int Getid(int x) {
    int k = x % modn;
    while (Hash[k][0] && Hash[k][0] != x) {
           k++;
           if (k == modn) k = 0;
    }
    if (!Hash[k][0]) Hash[k][1] = ++cnt;
    Hash[k][0] = x;
    return k;   
}

int Find(int x) {
    if (f[x] == x) return x;
    f[x] = Find(f[x]);
    return f[x];
}

int main() {
    int T, n;
    scanf("%d", &T);
    while (T--) {
           scanf("%d", &n);
           for (int i = 0; i < modn; i++) Hash[i][0] = 0, f[i] = i;
           cnt = 0;
           bool check = 1;
           for (int i = 1; i <= n; i++) {
                scanf("%d %d %d", &a[i].x, &a[i].y, &a[i].z);
                int idx = Getid(a[i].x);
                int idy = Getid(a[i].y);
                if (a[i].z == 1) {
                    int rpx = Find(idx);
                    int rpy = Find(idy);
                    f[rpx] = f[rpy];
                }
           }
           for (int i = 1; i <= n; i++) {
                if (!check) break;
                int idx = Getid(a[i].x);
                int idy = Getid(a[i].y);
                int rpx = Find(idx);
                int rpy = Find(idy);
                if (a[i].z == 1 && f[rpx] != f[rpy]) check = 0;
                if (a[i].z == 0 && f[rpx] == f[rpy]) check = 0; 
           }
           if (check) printf("YES\n");
                 else printf("NO\n");
    }
    return 0;
}
阅读更多
版权声明:欢迎借鉴,谢绝抄搬。 https://blog.csdn.net/Gx_Man_VIP/article/details/80498824
个人分类: C++ 并查集 哈希
想对作者说点什么? 我来说一句

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

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭