CF1400D Zigzags（思维）

Practice link： https://codeforces.ml/problemset/problem/1400/D

题意：给你一个数组，求（i、j、k、l) 满足

         的组合数量。

          数据范围： 4<=n<=3000   1<=a[ i ]<=n 

思路：数据范围很小，因此可以枚举 j 和 k ,只要求出位置 j 前面与 a[ k ] 相等的数目 x ，和位置 k 后面与 a[ j ]相等的数目 y ，最后答案加上x和y的乘积即可。

         用sum[ i ][ j ]来表示第 i 个数前面值为 j 的数的个数。在每次枚举中答案就是 ans=ans+sum[ j-1 ][ a[ k ] ] * (sum[ n ][ a[ j ] ] - sum[ k ][ a[ j ] ]) 。

代码：

 1 //#include<bits/stdc++.h>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<iostream>
 6 #include<string>
 7 #include<vector>
 8 #include<stack>
 9 #include<bitset>
10 #include<cstdlib>
11 #include<cmath>
12 #include<set>
13 #include<list>
14 #include<deque>
15 #include<map>
16 #include<queue>
17 #define ll long long
18 #define MOD 1000000007
19 #define INF 0x3f3f3f3f
20 #define mem(a,x) memset(a,x,sizeof(a))
21 #define lowbit(x) x&(-x)
22 #define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
23 using namespace std;
24  
25 int T;
26 int a[3005];
27 int sum[3005][3005];
28 int main()
29 {
30     int n;
31     cin>>T;
32     while(T--){
33         scanf("%d",&n);
34         mem(sum,0);
35         for(int i=1;i<=n;i++){
36             scanf("%d",&a[i]);
37             for(int j=1;j<=i;j++){
38                 sum[i][a[j]]++;
39             }
40         }
41         ll ans=0;
42         for(int j=2;j<=n-2;j++){
43             for(int k=j+1;k<=n-1;k++){
44                 ans=ans+1ll*sum[j-1][a[k]]*(sum[n][a[j]]-sum[k][a[j]]);
45             }
46         }
47         cout<<ans<<endl;
48     }
49     return 0;
50 }

 

posted @ 2020-08-27 15:28  hachuochuo  阅读(172)  评论(0)  编辑  收藏  举报