Codeforces Round #553 (Div. 2) E - Number of Components

CodeForces E题解
本文解析了CodeForces竞赛中一道题目E的解决方案,该题要求计算特定函数的累加和。通过优化的算法避免了O(n^3)的暴力求解,采用一次遍历判断相邻点权值,实现O(n)的高效解答。

题目链接:
http://codeforces.com/contest/1151/problem/E
题意:
n个点每个点和前一个点以及后一个点相连,且每个点有一个权值。定义一个函数f(l,r)权值在(l,r)的点的联通块的数量。求l从0-n,r从0-n的累加和。
思路:
引用一下出题人的原话
在这里插入图片描述
暴力求解很好想,枚举L,R的区间,在每个区间跑一次n循环,开一个新数组b记录权值在范围内的为1,其余为0。来看看不联通的1有几个。
当然这个复杂度是n3,我们可以跑一遍循环。判断相邻的两个点的权值,在多少种L和多少种R的情况下不连通。
如果a[i]>a[i-1]那么对于L要大于a[i-1]小于等于a[i],对于r要大于等于a[i]小于等于n;所以ans+=(a[i]-a[i-1])(n-a[i]+1)。
如果a[i]<a[i-1]那么对于L要大于0小于等于a[i],对于r要大于等于a[i]小于等于a[i-1]; 所以ans+=(a[i]-0)
(a[i-1]-a[i])。
代码:

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define for1(i,n) for(int i=1;i<=n;i++)
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
const int maxn = 1e5+5;

int a[maxn];

int main(){
    IO;
    int n;cin>>n;
    for1(i,n) cin>>a[i];
    ll ans = 0;
    for1(i,n){
        if(a[i]>a[i-1]) ans+=(a[i]-a[i-1])*1ll*(n-a[i]+1);
        if(a[i]<a[i-1]) ans+=a[i]*1ll*(a[i-1]-a[i]);
    }
    cout << ans<<'\n';
    return 0;
}

在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值