CodeForces 597C_Subsequences

Description
For the given sequence with n different elements find the number of increasing subsequences with k + 1 elements. It is guaranteed that the answer is not greater than 8·1018.
Input
First line contain two integer values n and k(1 ≤ n ≤ 105, 0 ≤ k ≤ 10) — the length of sequence and the number of elements in increasing subsequences.
Next n lines contains one integer ai (1 ≤ ai ≤ n) each — elements of sequence. All values ai are different.
Output
Print one integer — the answer to the problem.
n개수에서 길이가 k+1인 승차자열을 몇 개 찾는지 뜻한다.
dp[k][m]는 m로 끝나는 길이가 k인 하위 문자열의 수를 대표하며, 매번 업데이트할 때마다 dp[k-1][1]~dp[k-1][a[i]의 모든 값을 합치면 절대tle
처음에는 할 줄 몰랐는데 나중에 친구에게 물어본 후에 나무 모양의 그룹이 있다는 것을 알게 되었습니다. 추천:http://dongxicheng.org/structure/binary_indexed_tree/
#include
using namespace std;
typedef unsigned long long ull;
int n;
int k;
const int MAXN = 1e5 + 10;
const int MAXK = 15;
int a[MAXN];
ull dp[MAXK][MAXN];// n      k   
//    ------------------------
int lowbit(int t){
	return t&(t ^ (t - 1));
}
ull sum(int k,int end){
	ull sum = 0;
	while (end > 0){
		sum += dp[k][end];
		end -= lowbit(end);
	}
	return sum;
}
void pluss(int k, int pos, ull num){
	while (pos <= n){
		dp[k][pos] += num;
		pos += lowbit(pos);
	}
}
//--------------------------------
int main()
{
	while (cin >> n >> k){
		memset(dp, 0, sizeof(dp));
		for (int i = 1; i <= n; ++i){
			cin >> a[i];
			for (int tk = k + 1; tk >= 2; --tk){
				pluss(tk, a[i], sum(tk - 1, a[i]));
			}
			pluss(1, a[i], 1);
		}
		ull ans = 0;
		ans = sum(k + 1, n);
		cout << ans << endl;
	}
	return 0;
}

좋은 웹페이지 즐겨찾기