题意:n个节点的链,相邻节点有路,每个点一个权值,f(l,r)为权值在【l,r】的点的连通块数量求 思路:求每个节点作为连通块的最右边的节点的次数,区间【l,r】有连通块的最右边为a[i]则要满足条件,a[i]能满足区间[l,r],a[i+1]不能满足区间[l,r]; 就可以推出以下代码了;
for(int i = 1; i <= n; i++) { if(a[i] > a[i + 1]) ans += (a[i] - a[i + 1]) * (n - a[i] + 1); if(a[i] < a[i + 1]) ans += a[i] * (a[i + 1] - a[i]); }对于a[n],只需要把a[n+1]赋值为n+1即可 代码:
#include <bits/stdc++.h> using namespace std; #define ll long long const int maxn = 1e5 + 5; ll a[maxn]; int main() { // freopen("in.txt", "r", stdin); int n; cin >> n; for(int i = 1; i <= n; i++) cin >> a[i]; a[n + 1] = n + 1; ll ans = 0; for(int i = 1; i <= n; i++) { if(a[i] > a[i + 1]) ans += (a[i] - a[i + 1]) * (n - a[i] + 1); if(a[i] < a[i + 1]) ans += a[i] * (a[i + 1] - a[i]); } cout << ans << '\n'; return 0; }