題目鏈接：點擊查看

題目大意：給出一個長度為 n 的數列 a ，現在要求找到滿足條件的四元對 ( i , j , k , l ) 的個數：

1 <= i < j < k < l <= n

a[ i ] == a[ k ] && a[ j ] == a[ l ]

題目分析：n 只有 3000 ，顯然支持 n * n * logn 的時間復雜度，所以可以枚舉四元對中其中的兩個位置，然后去計算剩下的兩個位置中有多少個符合條件的，基于此，我們可以設計兩個簡單的 dp 輔助實現，dp1[ i ][ j ] 代表的是 [ 1 , i ] 區間內，j 出現的次數，相似的，dp2[ i ][ j ] 代表的是 [ i , n ] 區間內，j 出現的次數，這樣我們就可以 n * n 去枚舉 j 和 k ，對于每一對 j 和 k ，我們都可以直接得出 k 后面有多少個等于 a[ j ] 的個數，以及 j 前面有多少個等于 a[ k ] 的個數了，其乘積的累加和便是答案

代碼：
?

#include<iostream> #include<cstdio> #include<string> #include<ctime> #include<cmath> #include<cstring> #include<algorithm> #include<stack> #include<climits> #include<queue> #include<map> #include<set> #include<sstream> #include<cassert> #include<bitset> using namespace std;typedef long long LL;typedef unsigned long long ull;const int inf=0x3f3f3f3f;const int N=3e3+100;LL a[N],dp1[N][N],dp2[N][N];//dp1[i][j]:[1,i]內有多少個j,dp2[i][j]:[i,n]內有多少個j int main() { #ifndef ONLINE_JUDGE // freopen("data.in.txt","r",stdin); // freopen("data.out.txt","w",stdout); #endif // ios::sync_with_stdio(false);int w;cin>>w;while(w--){int n;scanf("%d",&n);for(int i=1;i<=n;i++)scanf("%lld",a+i);for(int i=1;i<=n;i++)dp1[0][i]=dp2[n+1][i]=0;for(int i=1;i<=n;i++){for(int j=1;j<=n;j++)dp1[i][j]=dp1[i-1][j];dp1[i][a[i]]++;}for(int i=n;i>=1;i--){for(int j=1;j<=n;j++)dp2[i][j]=dp2[i+1][j];dp2[i][a[i]]++;}LL ans=0;for(int j=1;j<=n;j++)for(int k=j+1;k<=n;k++)ans+=dp1[j-1][a[k]]*dp2[k+1][a[j]];printf("%lld\n",ans);}return 0; }

?

總結

以上是生活随笔為你收集整理的CodeForces - 1400D Zigzags(简单dp)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。