CodeForces-1373G. Pawns

传送门

题目大意：给大小为n*n的棋盘，第k列为特殊列，每个棋子只能向上、左上、右上走，m次放置棋子（给坐标），如果出现棋子坐标第二次出现则将棋子拿走，现在让所以棋子都走到特殊列（可能会溢出棋盘），求最高行最小超出n多少行，没超过就输出0，即求最终行最少溢出 n 多少行。

 

思路：线段树

（做的时候没想到用线段树维护唉，还以为每次修改都要推倒重来，其实看到修改就该往线段树那想了，只不过这里只有单点修改~

暴力tle，优化了两小时

首先求出每个棋子(x,y)，走到特殊列时的行数，abs(x-k)+y,也就是棋子放置的行至少是abs(x-k)+y，以下称为最小行

只问一次的话，很简单,只要将棋子从下往上摆放就行，有空格就放。

模拟一次可以发现，若第a行以下，可以容纳所有最小行＜a的棋子，那么最高行就从第a行开始算，即最高行至少为  a+最小行≥a的棋子数。

我们设最小行为第j行以及≥j的棋子数为f[ j ] ,设最终的最高行为h，根据f[j]的定义，恒有不等式 f[j] ≤ h - j + 1 ，移项得 h ≥ f[j] + j -1 ，接下来求最小的h使得式子恒成立，即求max(f[j] + j -1 )

假设每次修改第b行，放置棋子的话，区间[1,b]+1，拿走棋子的话[1,b]-1，接下来线段树维护一下就行，区间一开始赋值为j-1，然后每次修改就是改f[j]，最终线段树去维护区间最大的f[j] + j-1就行

注意一下线段树小细节就行：

1，lazy标别忘清

2，建树[1,2n]，棋子最大只可能落在2n，既然如此，内存也要开到8*n

#include <iostream>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <queue>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0)
#define _for(i,a,b) for(int i=(a) ;i<=(b) ;i++)
#define _rep(i,a,b) for(int i=(a) ;i>=(b) ;i--)
#define mst(x,y) memset(x,y,sizeof(x))
#define all(v) v.begin() , v.end()
#define pb(v) push_back(v)
#define INF 0x3f3f3f3f
#define int long long
#define ls p<<1
#define rs p<<1|1
#define lson p*2,l,mid
#define rson p*2+1,mid+1,r
typedef long long ll;
typedef pair<int ,int > pii;//坐标
int n,k,m;
const int N = 2e5+10;
const int mod = 1e9+7;
int tr[8*N];
int lz[8*N];
int cnt[2*N];
void pushup(int p)//维护最大 f[j]+j-1
{
    tr[p] = max(tr[ls] , tr[rs]);
}
void pushdown(int p)
{
    if( lz[p] )
    {
        int x=lz[p];
        lz[ls] += x,lz[rs] += x;
        tr[ls] += x,tr[rs] += x;
        lz[p]=0;
    }
    return;
}
void build(int p ,int l,int r)
{
    if(l==r)
    {
        tr[p] = l-1;//初始化赋值为j-1
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(p);
}
void update(int p,int l,int r ,int a ,int b ,int c)
{
    if( a<=l&&r<=b)
    {
        tr[p] += c;
        lz[p] += c;
        return;
    }
    if( lz[p] ) pushdown(p);
    int mid= (l+r)>>1;
    if(a<=mid)   update(lson,a,b,c);
    if(b>=mid+1) update(rson,a,b,c);
    pushup(p);
}
ll find(int p, int l ,int r ,int a ,int b)
{
    if( a<=l && r<=b )
    {
        return tr[p];
    }
    if( lz[p] ) pushdown(p);
    int mid=(l+r)>>1;
    int res=0;
    if( a<=mid) res = find(lson,a,b);
    if( b>=mid+1) res = max(res,find(rson,a,b));
    return res;
}
void solve()
{
    set <pii> p;
    set <int> mx;
    cin>>n>>k>>m;
    build(1,1,n<<1);//最多移动到2n的位置
    while( m-- )
    {
        int x,y;
        cin>>x>>y;
        int pos = abs( x-k ) + y;
        if( p.count({x,y}))//放棋子
       {
            cnt[pos]--;
            if( cnt[pos]==0 ) mx.erase(pos);
            p.erase({x,y});
            update(1,1,n<<1,1,pos,-1);
       }
       else//拿走棋子
       {
           cnt[pos]++;
           p.insert({x,y});;
           mx.insert(pos);
           update(1,1,n<<1,1,pos,1);
       }
       if(mx.empty() ) cout<<"0"<<endl;//防止集合为空
       else cout<<max(1ll*0,find(1,1,n<<1,1,*mx.rbegin())-n)<<endl;
    }

}
signed main()
{
    //!!!!!!!!!!!!!!!!!!!!!!
//    freopen("data.txt","r",stdin);
    //!!!!!!!!!!!!!!!!!!!!!!
    IOS;
    solve();
}