价值序列题解报告

价值序列题解报告

标签：极值 组合数

题目链接

来源：牛客网

解题思路：

根据题目定义，推导出一个单调序列的价值就是首位两端差的绝对值，那么先试着找出数列的极值点
为方便数据处理我们用两个数组来计入，
a数组计入去重后数组（这样可以方便判断是否为极值点）
b数组统计原数字个数
比如 1 3 3 4 4
数组 a 的值分别为 1 3 4
数组 b 的值分别为 1 2 2
随后分别考虑极值个数和非极值个数对最终答案的影响
对于连续相同的极值（上述的a[3]） 我们必须至少保留一项 ans * = 2的a[3]次-1
而对于全部非极值点都可以保留或不保留 ans * = 2*2
我们使用快速幂来减少复杂度

代码实现：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod = 998244353;
const int N = 1e5 + 8;
int a[N],c[N];
ll ksm(ll a, ll b) {
	ll ans = 1;
	while (b) {
		if (b & 1) (ans *= a) %= mod;
		(a *= a) %= mod;
		b >>= 1;
	}
	return ans;
}

bool check(int i){
	return (a[i] > a[i - 1] && a[i] > a[i + 1]) || (a[i] < a[i - 1] && a[i] < a[i + 1]);
}

int main() {
	int t;
	cin >> t;
	while (t--) {
		ll n;
		ll ans = 1;
		cin >> n;
		int na=0,x;
		memset(a,0,sizeof(a));
		memset(c,0,sizeof(c));
		for (int i = 1; i <= n; i++) {
			scanf("%d", &x);
			if(a[na]==x){
			   c[na]++;
			}
			else{
				na++;
				a[na]=x;
				c[na]++;
			}
					
		}
        
		int cnt=n;
		if(na==1){
			(ans *= ksm(2, c[1])-1) %= mod;
			cout<<ans<<endl;
			continue;
		}
		else{
			(ans *= ksm(2, c[1])-1) %= mod;
			(ans *= ksm(2, c[na])-1) %= mod;
			cnt -= c[1]+c[na];
		}
		
		for (int i = 2; i <= na-1; i++) {
			if (check(i)) {
			    cnt -= c[i];
				(ans *= ksm(2, c[i])-1) %= mod;	
			}
		}	

		(ans *= ksm(2, cnt)) %= mod;
		cout << ans << endl;
	}
	return 0;
}