Codeforces 1066B. Heaters（线段树区间修改+区间最小值）

题目：传送门

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <string>
#include <vector>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <algorithm>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0)
#define deg(x) cout<< #x"=" << x << endl;
#define degg(x,y) cout<< #x"="<< x <<","<< #y"="<< y <<endl;
#define deggg(x,y,z) cout<<#x"="<<x<<","<<#y"="<<y<<","<<#z"="<<z<<endl;
#define sca(x) scanf("%d",&x)
#define scaa(x,y) scanf("%d%d",&x,&y)
#define scaaa(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define pch(x) putchar(x)
#define ptf(x) printf("%d\n",x)
#define ptff(x,y) printf("%d %d\n",x,y)
#define ptfff(x,y,z) printf("%d %d %d\n",x,y,z)
#define space(x) printf("%d ",x)
#define mes(a,x) memset(a,x,sizeof(a))
#define PI acos(-1)
#define lowbit(x) x&-x
#define ALL(X) (X).begin(), (X).end()
#define PB push_back
#define MP make_pair
#define ls rt<<1
#define rs rt<<1|1
#define X first
#define Y second
using namespace std;

typedef long long ll;
typedef pair<int,int> pii;
typedef pair<ll,ll> pll;
typedef priority_queue<int,vector<int>,greater<int> > xqueue;
typedef priority_queue<int> dqueue;

const ll mod=1e9+7;
const ll INF=9e18;
const int inf=2147483647;

const int N=1e3+5;

int a[N];
int mn[N<<2],f[N<<2];

void maintain(int o,int l,int r)
{
    if(l==r) mn[o]=0;
    else mn[o]=min(mn[2*o],mn[2*o+1]);
    mn[o]+=f[o];
}

void add(int rt,int l,int r,int ql,int qr,int d)
{
    int m=(l+r)/2;
    if(ql<=l&&qr>=r) f[rt]+=d;
    else
    {
        if(ql<=m) add(ls,l,m,ql,qr,d);
        if(qr>m) add(rs,m+1,r,ql,qr,d);
    }
    maintain(rt,l,r);
}

int ask(int rt,int l,int r,int ql,int qr,int sum)
{
    int m=(l+r)/2;
    if(ql<=l&&qr>=r) return mn[rt]+sum;
    else
    {
        int ans=inf;
        sum+=f[rt];
        if(ql<=m) ans=min(ans,ask(ls,l,m,ql,qr,sum));
        if(qr>m) ans=min(ans,ask(rs,m+1,r,ql,qr,sum));
        return ans;
    }
}

void work()
{
    int n,r; scaa(n,r);
    int ans=0;
    for(int i=1;i<=n;i++)
    {
        sca(a[i]);
        if(a[i]) ans++;
    }
    for(int i=1;i<=n;i++)
    {
        if(a[i])
        {
            int l=max(1,i-r+1);
            int rr=min(n,i+r-1);
            add(1,1,n,l,rr,1);
        }
    }
    int t=ask(1,1,n,1,n,0);
    if(t<=0) ptf(-1);
    else
    {
        for(int i=1;i<=n;i++)
        {
            if(a[i])
            {
                int l=max(1,i-r+1);
                int rr=min(n,i+r-1);
                add(1,1,n,l,rr,-1);
                t=ask(1,1,n,l,rr,0);
                if(t<=0)
                    add(1,1,n,l,rr,1);
                else
                    ans--;
            }
        }
        ptf(ans);
    }
}

int main()
{
    work();
    return 0;
}