#树状数组#洛谷 4528 bzoj 1145 图腾

最新推荐文章于 2022-08-24 09:35:59 发布

lemondinosaur

最新推荐文章于 2022-08-24 09:35:59 发布

阅读量129

点赞数

分类专栏：树状数组文章标签：洛谷 4528 图腾 bzoj 1145 图腾

本文链接：https://blog.csdn.net/sugar_free_mint/article/details/98523843

版权

树状数组专栏收录该内容

31 篇文章 0 订阅

订阅专栏

题目

分析

$d p [1324] - d p [1243] - d p [1432]$ $= (d p [1 x 2 x] - d p [1423]) - (d p [12 x x] - d p [1234]) - (d p [14 x x] - d p [1423])$ $= d p [1 x 2 x] - d p [12 x x] + d p [1234] - d p [14 x x]$ $= d p [1234] + d p [1 x 2 x] + d p [13 x x] - d p [1 x x x]$
剩下的就是树状数组的事情了，首先可以用树状数组求出位置 $i$ 左右边比 $a_i$ 小的个数 $l [i], r [i]$

dp[1xxx]：显然为 $C_{r_i}^{3}$
dp[1234]：枚举2所在的位置 $i$ ，答案为 $(n-i-r[i])*\sum_{j<i,a[j]<a[i]}l[j]$ ，用树状数组维护
dp[1x2x]：枚举2所在的位置 $i$ ，右边有 $n - i - r [i]$ 种情况，左边需满足 $k < j < i, a [k] < a [i], a [j] > a [i]$
那么如果只考虑 $a [k] < a [i]$ 显然答案是 $l [i] * (i - 1)$ ，那么忽略了 $j < k < i 和 k < j < i, a [j] < a [i]$ 的情况，那么分别为 $C_{l[i]}^2$ 和 $\sum_{h<i,a[h]<a[i]}j$
所以答案为 $(n-i-r[i])*(l[i]*(i-1)-C_{l[i]}^2-\sum_{h<i,a[h]<a[i]}j)$
dp[13xx]：枚举3所在的位置 $i$ ，那么4就有 $n - i - r [i]$ 种情况，那么1和2要满足 $k < i < j, a [k] < a [j] < a [i]$
只考虑 $a [k] < a [i], a [j] < a [i], k < i$ 有 $l [i] (a [i] - 1)$ 种选法，需要减掉的是 $i\geq j$ 和 $a [k] > a [j]$ 的情况
那么也就分别是 $C_{l[i]}^2,\sum_{h<i,a[h]<a[i]}a[h]$
所以答案为 $(n-i-r[i])*(l[i]*(a[i]-1)-C_{l[i]}^2-\sum_{h<i,a[h]<a[i]}a[h])$

所以真的是一道思维好题呀

代码

#include <cstdio>
#include <cctype>
#include <cstring>
#define rr register
using namespace std;
const int mod=16777216;
int n,c[200101],l[200101],r[200101],a[200101];
inline signed iut(){
	rr int ans=0; rr char c=getchar();
	while (!isdigit(c)) c=getchar();
	while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
	return ans;
}
inline signed mo1(int x,int y){return x+y>=mod?x+y-mod:x+y;}
inline signed query(int x){rr int ans=0; for (;x;x-=-x&x) ans=mo1(ans,c[x]); return ans;}
inline void update(int x,int y){for (;x<=n;x+=-x&x) c[x]=mo1(c[x],y);}
signed main(){
	n=iut(); rr int ans=0;
	for (rr int i=1;i<=n;++i) a[i]=iut();
	for (rr int i=1;i<=n;++i) l[i]=query(a[i]),r[i]=a[i]-1-l[i],update(a[i],1);
	memset(c,0,sizeof(c));
	for (rr int i=1;i<=n;++i){//1xxx
		rr int x=n-i-r[i];
		ans=mo1(ans,mod-1ll*x*(x-1)*(x-2)/6%mod);
	}
	for (rr int i=1;i<=n;++i) ans=mo1(ans,1ll*(n-i-r[i])*query(a[i])%mod),update(a[i],l[i]);//1234
	memset(c,0,sizeof(c));
	for (rr int i=1;i<=n;++i) ans=mo1(ans,(1ll*l[i]*(i-1)%mod+mod-query(a[i])-1ll*l[i]*(l[i]-1)/2%mod)%mod*(n-i-r[i])%mod),update(a[i],i);//1x2x
    memset(c,0,sizeof(c));
    for (rr int i=n;i;--i) ans=mo1(ans,(query(a[i])+mod-1ll*r[i]*(r[i]+1)/2%mod)%mod*(n-i-r[i])%mod),update(a[i],a[i]);//13xx
    return !printf("%d",mo1(ans,mod));
}