CF:1626C. Monsters And Spells

题目链接

题解：

从后往前枚举，每个点所跨越的维度为p=k[i]-h[i]+1,即p为i点至少的起点。记录每个点的起点后。

从n-1点往前遍历；假如遍历到i点，判断i+1的起点是否覆盖了k[i]，若覆盖了，则i和i+1的起点为两者起点的最小值，后面同理。

所以相当于维护一些连通块。用并查集维护。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn=105;
int n;
int k[maxn];
int h[maxn];
int a[maxn],dmin[maxn];
int find(int x){
    if(x!=a[x]) a[x]=find(a[x]);
    return a[x];
}
void solve(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>k[i];
    for(int i=1;i<=n;i++) cin>>h[i];
    for(int i=1;i<=n;i++){
        a[i]=i;
        dmin[i]=k[i]-h[i]+1;
    }
    for(int i=n-1;i>=1;i--){
        int l=dmin[find(i)],r=dmin[find(i+1)];
        if(r<=k[i]){
            dmin[find(i+1)]=min(l,dmin[find(i+1)]);
            a[find(i)]=find(i+1);
        }
    }
    LL ans=0;
    for(int i=1;i<=n;i++){
        if(find(i)==i){
            int len=k[i]-dmin[i]+1;
            ans+=(LL)(len+1)*len/2;
        }
    }
    cout<<ans<<'\n';
}
int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t;cin>>t;
    while(t--) solve();
    return 0;
}