ABC186 F - Rook on Grid（树状数组）

题意：

解法：

显然要维护:
r[i]=第i行第一个障碍的下标,
c[i]=第i列第一个障碍的下标.

对于[1,c[1]-1]的每一行,显然答案累加r[i]-1,
对于[1,r[1]-1]的每一列,显然答案累加c[i]-1,
但是这样会统计到之前统计过的格子,考虑如何消去重复:
对于第i列,本来应该累加c[i]-1,
重复格子的个数就是1到c[i]-1行中r[]>i的行的数量,
那么不重复格子的数量就是1到c[i]-1行中r[]<=i的行的数量.
如何统计?
上面的条件是r[]<=i,发现i是递增的,
因此将(r[k],k)按照第一维r[]从小到大排序,
枚举列i的时候,只需要查询r[]<=i的(r[k],k)中,
有多少个满足k满足<=c[i]-1即可,
可以用树状数组存k,
查询的时候直接查询[1,c[i]-1]的前缀和即可.

code：

#include <bits/stdc++.h>
#define int long long
#define PI pair<int,int>
using namespace std;
const int maxm=2e6+5;
vector<int>g[maxm];
int r[maxm],c[maxm];
int n,m,w;
struct BIT{
    int c[maxm];
    int lowbit(int i){
        return i&-i;
    }
    void add(int i,int t){
        while(i<maxm)c[i]+=t,i+=lowbit(i);
    }
    int ask(int i){
        int ans=0;
        while(i)ans+=c[i],i-=lowbit(i);
        return ans;
    }
}T;
void solve(){
    cin>>n>>m>>w;
    for(int i=1;i<=n;i++){
        r[i]=m+1;
    }
    for(int i=1;i<=m;i++){
        c[i]=n+1;
    }
    for(int i=1;i<=w;i++){
        int x,y;cin>>x>>y;
        r[x]=min(r[x],y);
        c[y]=min(c[y],x);
    }
    int ans=0;
    //先向下再向右
    for(int i=1;i<=c[1]-1;i++){
        ans+=r[i]-1;
    }
    for(int i=c[1];i<=n;i++){
        r[i]=1;
    }
    //先向右再向下
    for(int i=1;i<=n;i++){//桶排序
        g[r[i]].push_back(i);
    }
    for(int i=1;i<=m;i++){
        if(c[i]==1)break;
        ans+=T.ask(c[i]-1);
        for(auto j:g[i]){
            T.add(j,1);
        }
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---