HU-3030 Incring Speed Limitsツリー配列

5109 ワード

http://acm.hdu.edu.cn/showproblem.php?pid=3030
この問題の意味は長さMのA行列を与えて、この配列に従って長さNの配列を作って、その後厳格に成長するサブシリアルの個数を求めます.
与えられた値を離散化して並べ替えると、ツリー配列で数を求めることができます.
コードは以下の通りです
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <map>
#include <algorithm>
#define MOD 1000000007
#define MAXN 500005
using namespace std;

int num[MAXN], A[MAXN], c[MAXN], seq[MAXN], cnt;

inline int lowbit(int x)
{
return x & -x;
}


inline void modify(int pos, int val)
{
for (int i = pos; i <= cnt; i += lowbit(i))
{
c[i] += val;
if (c[i] >= MOD)
c[i] %= MOD;
}
}

inline int getsum(int pos)
{
int s = 0;
for (int i = pos; i > 0; i -= lowbit(i))
{
s += c[i];
if (s >= MOD)
s %= MOD;
}
return s;
}

int main()
{
int T, n, m;
long long X, Y, Z, res;
scanf("%d", &T);
for (int ca = 1; ca <= T; ++ca)
{
res = 0;
map<int,int>mp;
scanf("%d %d %I64d %I64d %I64d", &n, &m, &X, &Y, &Z);
memset(c, 0, sizeof (c));
for (int i = 0; i < m; ++i)
{
scanf("%d", &A[i]);
}
for (int i = 0; i < n; ++i)
{
num[i+1] = A[i%m];
seq[i] = num[i+1];
A[i%m] = (X*A[i%m]+Y*(i+1))%Z;
}
sort(num+1, num+1+n);
cnt = unique(num+1, num+1+n) - (num+1);
for (int i = 1; i <= cnt; ++i)
{
mp[num[i]] = i;
}
modify(1, 1);
for (int i = 0; i < n; ++i)
{
long long x = getsum(mp[seq[i]]);
res += x;
if (res > MOD)
res %= MOD;
modify(mp[seq[i]]+1, x);
}
printf("Case #%d: %I64d
", ca, res);
}
return 0;
}