HDU 5517 【二维树状数组///三维偏序问题】

题目链接:【http://acm.split.hdu.edu.cn/showproblem.php?pid=5517】

题意:定义multi_set A<a , d>,B<c , d , e>,C<x , y , z>,给出 A , B ,定义 C = A * B = ={⟨a,c,d⟩∣⟨a,b⟩∈A, ⟨c,d,e⟩∈B and b=e} 。求出C之后,求C中一个元素t[i]<a , b , c>是否存在一个元素tmp<x , y , z>使得 x >= a&&y > =b&&z >= c && t != tmp;如果不存在ans++;输出ans即可。

题解:

  首先,我们要剪枝和去重,加入集合A中存在<1 , 3>,<2, 3>,<3, 3> ,<4, 3>,<4, 3>,那么只需要保留<4,3>就可以了,并且记录一下数量。因为前面的数都可以找到比它大的元素。求出C之后,我们依旧要对C去重。那么这道题就是求C某个元素是否存在大于它的元素了。因为这道题中B set里面c,d,很小在生成的C中y,z也很小,都小于1000,那么我们就可以用二维树状数组做了。首先对C进行从大到小排序,树状数组中只记录y,z的值。我们枚举C中的额每一个元素,先查询(1001 - y ,1001 - z),然后在更新(1001 - y ,1001 -  z),应为是从大到小排序的,所以后面的元素的x值一定小于等于前面的元素的x值,s如果查询不为0,那么一定存在某个元素是大于它的。我们一可以直接套CDQ分治的三维偏序模板,为维护值即可。

树状数组:

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 1e5 + 15;
int ic, N, M;
struct Point
{
int x, y, z;
LL w;
Point() {}
Point(int x, int y, int z, LL w): x(x), y(y), z(z), w(w) {}
bool operator < (const Point& T) const
{
if(x != T.x) return T.x < x;
if(y != T.y) return T.y < y;
return T.z < z;
}
bool operator == (const Point T) const
{
return (x == T.x && y == T.y && z == T.z);
}
} P[maxn];
int T[maxn], C[maxn];
int Tbit[1050][1050];
int low_bit(int x)
{
return x & -x;
}
void Bit_add(int x, int y, int val)
{
for(int i = x; i <= 1001; i += low_bit(i))
for(int j = y; j <= 1001; j += low_bit(j))
Tbit[i][j] += val;
}
int Bit_sum(int x, int y)
{
int ret = 0;
for(int i = x; i; i -= low_bit(i))
for(int j = y; j; j -= low_bit(j))
ret += Tbit[i][j];
return ret;
}
int main ()
{
scanf("%d", &ic);
for(int cs = 1; cs <= ic; cs++)
{
scanf("%d %d", &N, &M);
memset(T, 0, sizeof(T));
for(int i = 1; i <= N; i++)
{
int a, b;
scanf("%d %d", &a, &b);
if(a > T[b]) T[b] = a, C[b] = 1;
else if(a == T[b]) C[b]++;
}
int tmp = 0;
for(int i = 1; i <= M; i++)
{
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
if(T[c]) P[++tmp] = Point(T[c], a, b, (LL)C[c]);
}
sort(P + 1, P + 1 + tmp);
N = 1;
for(int i = 2; i <= tmp; i++)
{
if(P[i] == P[N]) P[N].w += P[i].w;
else
P[++N] = P[i];
}
LL ans = 0;
memset(Tbit, 0, sizeof(Tbit));
for(int i = 1; i <= N; i++)
{
if(!Bit_sum(1001 - P[i].y, 1001 - P[i].z)) ans += P[i].w;
Bit_add(1001 - P[i].y, 1001 - P[i].z, 1);
}
printf("Case #%d: %lld\n", cs, ans);
}
return 0;
}

三维偏序:

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 1e5 + 15;
int cs, N, M;
int T[maxn], C[maxn];
LL W[maxn], ans[maxn];
struct Edge
{
int x, y, z, id;
LL w;
Edge() {}
Edge(int x, int y, int z, int id, LL w): x(x), y(y), z(z), id(id), w(w) {}
bool operator < (const Edge & T) const
{
return z < T.z;
}
bool operator == (const Edge & T) const
{
return T.x == x && T.y == y && T.z == z;
}
} B[maxn], p[maxn], t[maxn];
bool cmpx(Edge &a, Edge &b)
{
if(a.x == b.x && a.y == b.y && a.z == b.z)
return a.id < b.id;
else if(a.x == b.x && a.y == b.y)
return a.z < b.z;
else if(a.x == b.x)
return a.y < b.y;
else
return a.x < b.x;
}
bool cmpy(Edge &a, Edge &b)
{
if( a.y == b.y && a.z == b.z)
return a.id < b.id;
else if(a.y == b.y)
return a.z < b.z;
else
return a.y < b.y;
}
//--------------------------------------------------------//树状数组
int c[maxn], maxz = 1001;
inline int lowbit(int x)
{
return x & (-x);
}
void add(int x, int val)
{
while(x <= maxz) c[x] += val, x += lowbit(x);
}
int Get_sum(int x)
{
int ret = 0;
while(x > 0) ret += c[x], x -= lowbit(x);
return ret;
}
//-------------------------------------------------
void cdq(int l, int r)
{
if(l == r) return ;
int mid = (l + r) / 2;
cdq(l, mid), cdq(mid + 1, r);
for(int i = l; i <= r; i++) t[i] = p[i];
sort(t + l, t + mid + 1, cmpy), sort(t + mid + 1, t + r + 1, cmpy);
int a1 = l, a2 = mid + 1;
for(int i = l; i <= r; i++)
{
if(a1 == mid + 1)
ans[t[a2].id] += Get_sum(t[a2].z), a2++;
else if(a2 == r + 1)
add(t[a1].z, 1), a1++;
else if(t[a1].y <= t[a2].y)
add(t[a1].z, 1), a1++;
else
ans[t[a2].id] += Get_sum(t[a2].z), a2++;
}
for(int i = l; i <= mid; i++) add(t[i].z, -1);
}
int main ()
{
scanf("%d", &cs);
for(int ic = 1; ic <= cs; ic++)
{
memset(T, 0, sizeof(T));
memset(ans, 0, sizeof(ans));
scanf("%d %d", &N, &M);
for(int i = 1; i <= N; i++)
{
int a, b;
scanf("%d %d", &a, &b);
if(a > T[b]) T[b] = a, C[b] = 1;
else if(a == T[b]) C[b]++;
}
int tmp = 0;
for(int i = 1; i <= M; i++)
{
int a, b, c;
scanf("%d %d %d", &a, &b, &c);
if(T[c]) ++tmp, p[tmp] = Edge(T[c], a, b, i, (LL)C[c]);
}
sort(p + 1, p + 1 + tmp, cmpx);
N = 1;
for(int i = 2; i <= tmp; i++)
{
if(p[i] == p[N]) p[N].w += p[i].w;
else p[++N] = p[i];
}
if(N <= 1)
{
printf("Case #%d: 0\n", ++ic);
continue;
}
for(int i = 1; i <= N; i++)
{
p[i].x = 100001 - p[i].x;
p[i].y = 1001 - p[i].y;
p[i].z = 1001 - p[i].z;
}
for(int i = 1; i <= N; i++)
{
p[i].id = i;
W[i] = p[i].w;
}
sort(p + 1, p + 1 + N, cmpx);
cdq(1, N);
p[N + 1].x = -1;
for(int i = N; i >= 1; i--)
{
if(p[i].x == p[i + 1].x && p[i].y == p[i + 1].y && p[i].z == p[i + 1].z)
ans[p[i].id] = ans[p[i + 1].id];
}
LL ret = 0;
for(int i = 1; i <= N; i++) if(!ans[i]) ret += W[i];
printf("Case #%d: %lld\n", ic, ret);
}
return 0;
}

  

上一篇:Anaconda的安装和更新


下一篇:HDU 4456(二维树状数组+坐标转换)