“蔚来杯“2022牛客暑期多校训练营9 题解

A题 Car show
思路：利用双指针，固定每次搜索的左指针向右搜索每个可能的区间，只要搜到了一个符合的区间，那这个区间里再增加元素也一定是符合要求的，停止搜索以减少复杂度。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+3;
ll n,m;
ll a[N],vis[N];
int main(){
    scanf("%lld%lld",&n,&m);
    set<int> s;//用集合的去重特性记录当前搜索区间内的元素，当元素个数等于m的时候，说明数字集齐了
    for(int i=0;i<n;i++){
        scanf("%lld",&a[i]);
    }
    ll ans=0;
    for(int i=0,j=0;i<=n-m;i++){//n-m往后一定没有满足的子区间
        while(s.size()!=m) {
            vis[a[j]]++;//vis数组用于记录当前区间内对应的数字有多少个
            s.insert(a[j]);
            j++;
            if(j>=n) break; 
        }
        if(s.size()==m) ans+=n-j+1;
        else break;//当j搜索到最后都没有符合的子区间，说明i再往后也不会有了
        vis[a[i]]--;//i位置上的元素挪出当前的搜索区间
        if(vis[a[i]]==0) s.erase(a[i]);
    }
    printf("%lld\n",ans);
}

B题 Two Frogs
思路：一共有n片荷叶，两只青蛙要从第一片荷叶开始跳到最后一片荷叶的位置，给出青蛙在第i片荷叶能跳的最远距离ai，保证i+ai<=n。求两只青蛙跳的次数相同的期望。由于青蛙有很多种不同的跳法都可能导致步数相同，青蛙在荷叶i上跳到之后的ai片荷叶上的概率都是相等的1/ai，基本的思路就是将步数相同的情况的概率两两相乘再相加，用逆元处理分数模的情况。有一个很妙的情况是，两两相乘再相加的情况可以转换成平方和的形式，这样就大大降低了时间复杂度。

#include<bits/stdc++.h>
using namespace std;
const int N=8001;
typedef long long ll;
const int mod=998244353;
ll f[N][N],sum[N];
ll a[N];
ll qk(ll a,ll b){
	ll ans=1;
	a%=mod;
	b%=mod;
	while(b){
		if(b&1) ans=ans*a%mod;
		a=a*a%mod;
		b>>=1;
		b%=mod;
	}
	return ans;
}
ll inv(int x){
	return qk(x,mod-2);
}

int main(){
	int n;
	cin>>n;
	for(int i=1;i<=n-1;i++){
		cin>>a[i];
	}
	f[0][1]=1;
	//fij的意思是，用i步跳到j上的概率
	for(int i=1;i<n;i++){
		for(int j=1;j<n;j++){
			//sum是一个差分数组，要对差分数组的每一个元素进行加减操作的话，只需要对它的开头和末尾元素进行操作即可
			sum[j+1]=(sum[j+1]+f[i-1][j]*inv(a[j])%mod)%mod;
			sum[j+1+a[j]]=(sum[j+1+a[j]]+mod-f[i-1][j]*inv(a[j])%mod)%mod;
		}
		//计算用i步跳到j的概率之和
		for(int j=1;j<=n;j++){
			f[i][j]=(f[i][j]+f[i][j-1]+sum[j])%mod;
		}
		memset(sum,0,sizeof(sum));
	}
	int ans=0;
	for(int i=0;i<=n;i++){
		ans=(ans+f[i][n]*f[i][n]%mod)%mod;
	}
	cout<<ans<<endl;
}