Atcoder ABC390F Double Sum 3 题解 [ 绿 ] [ 贡献思维 ] [ 计数 ]

Double Sum 3:简单计数题。

思路

首先考虑单个区间的 f 值如何计算,显然等于值域上连续段的个数。那么我们进一步观察值域上连续段的性质,发现一个连续段的开头一定满足比开头小 1 的数字不存在于区间中的特点。

于是我们就可以考虑对每一个数字计算贡献了。同样,一个数字 a 的贡献只有当区间包含 a 且不包含 a1 的时候才为 1,其他都是 0。因为只有这样才能组成一个连续段的开头,而我们的贡献就是以开头来计数的。当然这里以连续段结尾来计数也没有问题。

因此只需要对每个数计算包含 a 且不包含 a1 的区间个数即可。具体实现上,我们可以前后扫一遍,维护一个桶,包含 a 的区间我们钦定让最后面的那个 a 计算贡献,然后排除掉包含 a1 的区间即可。假设 a 前面的最后一个 a1prea1 处,后面的第一个 asufa 处,后面的第一个 a1sufa1 处,则贡献即为 (iprea1)×(min(sufa,sufa1)i)

时间复杂度 O(n)

代码

#include <bits/stdc++.h>
#define fi first
#define se second
#define lc (p<<1)
#define rc ((p<<1)|1)
#define eb(x) emplace_back(x)
#define pb(x) push_back(x)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef long double ldb;
using pi=pair<int,int>;
int n,a[300005],tot[300005],pre[300005];
ll ans=0;
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++)cin>>a[i];
    memset(tot,0,sizeof(tot));
    for(int i=1;i<=n;i++)
    {
        tot[a[i]]=i;
        pre[i]=tot[a[i]-1];
    }
    fill(tot,tot+n+1,n+1);
    for(int i=n;i>=1;i--)
    {
        int suf=min(tot[a[i]],tot[a[i]-1]);
        ans+=1ll*(i-pre[i])*(suf-i);
        tot[a[i]]=i;
    }
    cout<<ans;
    return 0;
}
posted @   KS_Fszha  阅读(22)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 微软正式发布.NET 10 Preview 1:开启下一代开发框架新篇章
· 没有源码,如何修改代码逻辑?
· PowerShell开发游戏 · 打蜜蜂
· 在鹅厂做java开发是什么体验
· WPF到Web的无缝过渡:英雄联盟客户端的OpenSilver迁移实战
点击右上角即可分享
微信分享提示