题目链接:
题目大意:
给出一个序列,问能选出多少个长度为三的子序列,满足后一项除以前一项为k。
题目分析:
我利用一个map类型的dp[i][v]数组,用来记录长度为i的子序列以v结尾的有多少种。很容易得到一个递推式:
dp[i][v] = dp[i-1][v/k]
dp[1][v] = 1;
AC代码:
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <map>
#define MAX 200007
using namespace std;
typedef long long LL;
int n,k;
int a[MAX];
map<int,LL> dp[4];
LL ans;
int main ( )
{
while (~scanf ( "%d%d" , &n , &k ) )
{
ans = 0;
for ( int i = 0 ; i < 4 ; i++ )
dp[i].clear();
for ( int i = 1 ; i <= n ; i++ )
scanf ( "%d" , &a[i] );
for ( int i = 1 ; i <= n ; i++ )
{
int x = a[i];
if ( x%k == 0 )
{
dp[3][x] += dp[2][x/k];
dp[2][x] += dp[1][x/k];
}
dp[1][x] += 1;
}
map<int,LL>::iterator it = dp[3].begin();
for ( ; it != dp[3].end() ; it++ )
ans += (*it).second;
printf ( "%I64d\n" , ans );
}
}