SCUT129 笔芯值 —— DP or 容斥原理

xiaoxiao2021-02-28  113

题目链接:https://scut.online/p/129

题解:

因为ai<=100000, 且ai只出现一次, 所以可以用pos[]数组保存ai的下标, 这样就不用向前遍历查找了,做到一步访问。 如果ai很大但仍只出现一次,亦可以用map存其下标。

设dp[i]为:以a[i]为结尾的所有子序列的bx值之和。

DP;

1假设已经求出了dp[i-1],那么在a[1]~a[i-1]的基础上,再加上a[i]。先假设a[i]的存在并不能对bx值作出贡献, 即以a[i]为结尾的所有子序列的bx值之和dp[i] = dp[i-1] 。

2.经过分析,可知:以a[i]为子序列的结尾,往前溯。 当没有出现a[i]-1或者a[i]+1时,a[i]对当前的序列都贡献了1,所以每往前一步,dp[i]都能增加1。当出现a[i]+1 or a[i]-1其中的一个时, 每往前 溯一步, a[i]的作用无效或者拆散了一对又合成了一对,所以在这一区间,dp[i]无变化。 当a[i]+1 and a[i]-1都出现时, 每往前溯一步, a[i]都拆散了一对,所以dp[i]要减去1。

最后答案就是dp[i]之和。

代码如下:

#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <string> #include <vector> #include <map> #include <set> #include <queue> #include <stack> #include <sstream> #include <algorithm> using namespace std; #define pb push_back #define mp make_pair #define ms(a, b) memset((a), (b), sizeof(a)) #define eps 0.0000001 typedef long long LL; const int INF = 2e9; const LL LNF = 9e18; const int mod = 1e9+7; const int maxn = 100000+10; LL a[maxn], pos[maxn], dp[maxn]; int n; void init() { scanf("%d",&n); for(int i = 1; i<=n; i++) scanf("%lld",&a[i]); ms(pos,0); ms(dp,0); } void solve() { int p; pos[a[1]] = 1; for(int i = 2; i<=n; i++) { dp[i] = dp[i-1]; //先假设a[i]的存在对bx值没有贡献。 p = max(pos[a[i]-1], pos[a[i]+1]); //找到a[i]+1 or a[i]-1 的下标, 都不存在时, p为0。 dp[i] += i-p-1; //对于以a[i]为结尾的,且不存在a[i]+1 or a[i]-1的子序列,a[i]的存在能增加一对,这样的序列有i-p-1对。 //由于只出现一个时, 对dp[i]没有影响, 所以跳过 //如果a[i]+1 and a[i]-1 都存在 if(pos[a[i]-1] && pos[a[i]+1]) { p = min(pos[a[i]-1], pos[a[i]+1]); //找到下标最小的哪一个 dp[i] -= p; //对于以a[i]为结尾的,且a[i]+1 and a[i]-1都存在的子序列,a[i]的存在拆散了一对,这样的序列有p对。 } pos[a[i]] = i; //记录a[i]出现的位置 } LL ans = 0; for(int i = 1; i<=n; i++) ans += dp[i]; printf("%lld\n",ans); } int main() { int T; scanf("%d",&T); while(T--){ init(); solve(); } }

容斥:

#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <string> #include <vector> #include <map> #include <set> #include <queue> #include <stack> #include <sstream> #include <algorithm> using namespace std; #define pb push_back #define mp make_pair #define ms(a, b) memset((a), (b), sizeof(a)) #define eps 0.0000001 typedef long long LL; const int INF = 2e9; const LL LNF = 9e18; const int mod = 1e9+7; const int maxn = 100000+10; LL a[maxn], pos[maxn], dp[maxn]; int n; void init() { scanf("%d",&n); for(int i = 1; i<=n; i++) scanf("%lld",&a[i]), pos[a[i]] = i; } void solve() { LL ans = 0; for(int len = 2; len<=n; len++) ans += 1LL*(len-1)*(n-len+1); for(int i = 1; i<n; i++) { int pre = pos[i]; int next = pos[i+1]; if(pre>next) swap(pre, next); ans -= 1LL*pre*(n-next+1); } printf("%lld\n",ans); } int main() { int T; scanf("%d",&T); while(T--){ init(); solve(); } }

转载请注明原文地址: https://www.6miu.com/read-27526.html

最新回复(0)