結果

問題 No.992 最長増加部分列の数え上げ
ユーザー square1001
提出日時 2020-02-14 21:46:19
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,278 ms / 2,000 ms
コード長 1,785 bytes
コンパイル時間 1,112 ms
コンパイル使用メモリ 88,100 KB
実行使用メモリ 99,456 KB
最終ジャッジ日時 2024-10-06 11:11:25
合計ジャッジ時間 30,216 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <map>
#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
const int mod = 1000000007;
const int inf = 1012345678;
int sz, seg[524295];
void update(int pos, int val) {
	pos += sz;
	seg[pos] = val;
	while (pos > 1) {
		pos >>= 1;
		seg[pos] = max(seg[pos * 2], seg[pos * 2 + 1]);
	}
}
int query(int a, int b, int k, int l, int r) {
	if (r <= a || b <= l) return -inf;
	if (a <= l && r <= b) return seg[k];
	int lc = query(a, b, k * 2, l, (l + r) >> 1);
	int rc = query(a, b, k * 2 + 1, (l + r) >> 1, r);
	return max(lc, rc);
}
map<int, int> bit[200009];
void add(int f, int x, int val) {
	for (int i = x + 1; i <= sz; i += i & (-i)) {
		bit[f][i] += val;
		if (bit[f][i] >= mod) bit[f][i] -= mod;
	}
}
int sum(int f, int r) {
	int ans = 0;
	for (int i = r; i >= 1; i -= i & (-i)) {
		ans += bit[f][i];
		if (ans >= mod) ans -= mod;
	}
	return ans;
}
int main() {
	int N;
	cin >> N;
	vector<int> A(N);
	for (int i = 0; i < N; ++i) {
		cin >> A[i];
	}
	vector<int> perm(N);
	for (int i = 0; i < N; ++i) {
		perm[i] = i;
	}
	sort(perm.begin(), perm.end(), [&](int i, int j) { return A[i] != A[j] ? A[i] < A[j] : i > j; });
	sz = 1;
	while (sz < N) sz *= 2;
	vector<int> dp(N);
	for (int i = 0; i < N; ++i) {
		dp[perm[i]] = query(0, perm[i], 1, 0, sz) + 1;
		if (dp[perm[i]] <= 0) dp[perm[i]] = 1;
		update(perm[i], dp[perm[i]]);
	}
	vector<int> ways(N);
	for (int i = 0; i < N; ++i) {
		if (dp[perm[i]] == 1) {
			ways[perm[i]] = 1;
		}
		else {
			ways[perm[i]] = sum(dp[perm[i]] - 1, perm[i]);
		}
		add(dp[perm[i]], perm[i], ways[perm[i]]);
	}
	int len = *max_element(dp.begin(), dp.end());
	int ans = 0;
	for (int i = 0; i < N; ++i) {
		if (dp[i] == len) {
			ans += ways[i];
			if (ans >= mod) ans -= mod;
		}
	}
	cout << ans << endl;
	return 0;
}
0