哞叫时间Ⅱ

合集 - 蓝桥杯训练(22)

1.路径之谜02-172.3.赢球票02-183.哞叫时间02-184.蛋糕游戏02-195.5.Excel地址02-20

6.哞叫时间Ⅱ02-20

7.6.k倍区间02-208.6135.奶牛体检02-219.11. 青蛙跳杯子02-2110.农夫约翰最喜欢的操作02-2211.蓝桥2013·带分数02-2312.拐杖糖盛宴02-2513.5438.密病牛02-2614.5439.约翰种地02-2715.5524.多数意见02-2816.炮弹03-0117.5526.细菌平衡03-0418.重复字符串03-0419.5538.回文游戏03-0520.5539.牛奶交换03-0621.5540.最大程度提高生产力03-0722.5590. 沿栅栏散步03-11

收起

题目链接：https://www.acwing.com/problem/content/description/6137/

题意：

给定一个长度为n的数组，求出三个数ai，ap，aq（ai！=ap，ap=aq）（i<p<q)的种类数量

思路：

时间复杂度需O（n），因此程序只能有一重循环搞定
不妨从后向前遍历，如果一个数字出现次数==两次，让计数器k++。
由于我们需要种类不能重复，且要满足122式的条件。所以用res记录（i，n】的出现次数大于等于2的数字总数，用ai去配。
那么不难发现，如果ai本身出现大于等于2次，需要将ai ai ai这种情况剔除一次，如果ai遍历过了，那么只需要看现在遍历的位置和ai上次出现位置之间，有哪些新的数出现次数大于等于2次加上去即可
可能表达的不是很清楚，希望代码更容易让人懂一些

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
#define int long long
#define endl "\n"
#pragma GCC optimize(3)
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const ll llmax=LLONG_MAX;
const int maxn=1e6+5;
const int mod=1e9+7;
int n;
int a[maxn];
int tot[maxn];
int ans;
int res[maxn];
int k;
bool vis[maxn];
pii pos[maxn];
 
signed main()
{
	ios::sync_with_stdio(false),cin.tie(0);
	cin>>n;
	rep(i,1,n) cin>>a[i];
	
	for(int i=n;i>=1;i--){
		tot[a[i]]++;
		res[i]=k;
		if(tot[a[i]]==2){
			k++;	
		}
		if(i<=n-2){
		if(!vis[a[i]]){
			ans+=res[i];
			if(tot[a[i]]>2&&pos[a[i]].second==0){
				ans--;
				pos[a[i]].second=1;
			}
			vis[a[i]]=true;
			pos[a[i]].first=i;
		}else{
			ans+=(res[i]-res[pos[a[i]].first]);
			if(tot[a[i]]>2&&pos[a[i]].second==0){
				ans--;pos[a[i]].second=1;
			}
			pos[a[i]].first=i;
		}
	}
	}
	
	cout<<ans;
	return 0;
}