1. 程式人生 > 實用技巧 >CF1400D Zigzags(思維)

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 }