HDU 6992 Lawn of the Dead (线段树)

最新推荐文章于 2022-02-25 22:07:09 发布

原创最新推荐文章于 2022-02-25 22:07:09 发布 · 339 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

数据结构专栏收录该内容

27 篇文章

订阅专栏

本文介绍了一种使用线段树数据结构解决大小为n*m的网格图可达性问题的方法。通过维护每一行可达点，避开k个禁止行走的点，计算从起点(1,1)出发的可达点数量。代码展示了如何运用线段树进行高效更新和查询操作。

链接： Lawn of the Dead

题意：

有一个大小为 $m(1\leq n,m\leq1e5)$ 的网格图，初始在点 $(1, 1)$ ，只能向下和向右走，现给出 k 的点的坐标，代表这 k 个点不能走，求图中有多少个点可到达。

思路：

如果一个点不可达，那么它上方和左边的点一定不可达，所以我们从上到下维护每一行可达的点有哪些，对于如果当前在第 i 行，点 $a_{i ,j}$ 和点 $a_{i,k}$ 是不能走的，那么我们就找到第 i - 1 行中区间[j + 1 , k - 1]内最左边的能到达的点 x，那么第i行的 [x , k - 1]就是可到达的。用线段树修改一下就好了。

代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6 + 5;
const double Pi = acos(-1.0);
int T;
int n , m , k;
vector<int>vec[maxn];
int sum[3][maxn << 2] , la[3][maxn << 2];

void pushup(int id , int rt){
    sum[id][rt] = sum[id][rt << 1] + sum[id][rt << 1 | 1];
}
void pushdown(int id, int rt , int l , int r){
    if(la[id][rt] == -1) return ;
    int mid = (l + r) / 2;
    sum[id][rt << 1] = (mid - l + 1) * la[id][rt];
    sum[id][rt << 1 | 1] = (r - mid) * la[id][rt];
    la[id][rt << 1] = la[id][rt];
    la[id][rt << 1 | 1] = la[id][rt];
    la[id][rt] = -1;
}
void build(int id , int l , int r , int rt){
    la[id][rt] = -1;
    if(l == r){
        sum[id][rt] = 0;
        return ;
    }
    int mid = (l + r) / 2;
    build(id , l , mid , rt << 1);
    build(id , mid + 1 , r , rt << 1| 1);
    pushup(id , rt);
}
void update(int id , int L , int R , int val, int l , int r , int rt){
    if(L <= l && R >= r){
        sum[id][rt] = (r - l + 1) * val;
        la[id][rt] = (r - l + 1) * val;
        return ;
    }
    pushdown(id , rt , l , r);
    int mid = (l + r) / 2;
    if(L <= mid) update(id , L , R ,val , l , mid , rt << 1);
    if(R > mid) update(id , L , R ,val ,  mid + 1 , r , rt << 1 | 1);
    pushup(id , rt);

}
int query(int id , int L , int R, int l ,int r , int rt){
    if(l == r){
        if(sum[rt]) return l;
        return -1;
    }
    pushdown(id , rt , l , r);
    int mid = (l + r) / 2;
    int pos = -1;
    if(L <= mid && sum[id][rt << 1]) pos =  query(id , L , R , l , mid , rt << 1);        //先找左边  没找到再找右边  ，不能不找
    if(pos != -1) return pos;
    if(R > mid && sum[id][rt << 1 | 1]) return query(id , L , R , mid + 1 , r , rt << 1 | 1);
    return -1;
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d",&n,&m,&k);
        ll ans = 0;
        for(int i = 1; i <= n; i ++) vec[i].clear();
        for(int i = 1,x , y; i <= k; i ++){
            scanf("%d%d",&x,&y);
            vec[x].push_back(y);
        }
        for(int i = 1; i <= n; i ++){
            vec[i].push_back(0);
            vec[i].push_back(m + 1);
            sort(vec[i].begin() , vec[i].end());
        }
        int now = 1;
        build(0 , 1 , m , 1);
        build(1 , 1 , m , 1);
        
        if(vec[1].size() == 2){
            update(0 , 1 , m ,1 ,  1 , m , 1);
            ans += m;
        }
        else{
            update(0 , 1 , vec[1][1] - 1 , 1 ,  1 , m , 1);
            ans += vec[1][1] - 1;
        }
        
        for(int i = 2; i <= n; i ++){
            for(int j = 0; j < vec[i].size() - 1; j ++){
                int l = vec[i][j] + 1 , r = vec[i][j + 1] - 1;
                if(l > r) continue;
                
                int pos = query(now ^ 1 , l , r , 1 , m , 1);
                if(pos == - 1) continue;                              //上方没有可行点
                
                update(now , pos , r , 1 , 1 , m , 1);
                ans += r - pos + 1;
            }
            now ^= 1;
            update(now , 1, m , 0 , 1 , m , 1);
        }
        printf ("%lld\n",ans);
    }

    return 0;
}