Codeforces Round 946 (Div. 3)C. Beautiful Triple Pairs

最新推荐文章于 2024-06-12 15:24:14 发布

lzltxdy

最新推荐文章于 2024-06-12 15:24:14 发布

阅读量427

点赞数 8

文章标签： c语言算法数据结构

本文链接：https://blog.csdn.net/lzltxdy/article/details/139099748

版权

题目描述:

给你一个长度为n的数组a,请你求出所有三元组 $a_j,a_{j+1},a_{j+2}]$ 中完美三元组的对数
$设有三元组b,c\\ 完美三元组的定义: 满足其中一个即可\\ 1.b_1\neq c_1 和 b_2 = c_2 和 b_3=c_3\\ 2.b_1= c_1 和 b_2 \neq c_2 和 b_3=c_3\\ 3.b_1=c_1 和 b_2 = c_2 和 b_3 \neq c_3\\$

思路:

$哈希记录一下就行了$

代码:

#include <bits/stdc++.h>

using namespace std;

using ll=long long;
int t;

int main(){
    cin >> t;
    while(t--){
        int n;cin >> n;
        vector<int> a(n,0);
        for(int i=0;i<n;i++) cin >> a[i];
        map<array<int,3>,int> cnt;//建立三元组哈希
        ll res=0;
        for(int i=0;i<n-2;i++){
        	//记录此时的三元组
            array<int,3> th={a[i],a[i+1],a[i+2]};
            //考虑所有可能相同的情况
            vector<array<int,3>> list={
                {0,a[i+1],a[i+2]},
                {a[i],0,a[i+2]},
                {a[i],a[i+1],0}
            };
            for(auto i:list){
            	//该三元组对答案的贡献为"该情况的数目"-"和该三元组相同的数目"
                res+=cnt[i]-cnt[th];
                cnt[i]++;//该情况的数目加一
            }
            cnt[th]++;//该三元组的数目加一
        }
        cout << res << endl;
    }
    return 0;
}