题目链接:http://codeforces.com/contest/786/problem/C
题意:给了一个长度为n的数列,现在要问你对于每个k,k从取到n,最少可以把n分成多少段,使得每一段里面的颜色总数<=k。
解法:
所以我们倒着建主席树,然后在主席树上二分就可以对于每一个k用两个log查询到答案了。顺着建立主席树的话,你还要二分右端点,这样的话是3个log,不知道是否能过,稍加转化就是2个log了。这是在线的做法。
//CF 787E
#include <bits/stdc++.h>
using namespace std;
const int maxn = 110000;
int a[maxn];
struct node{
int l, r, sum;
}T[40*maxn];
int root[maxn], pre[maxn], cnt, n;
void update(int l, int r, int &x, int y, int pos, int k){
T[++cnt] = T[y], T[cnt].sum += k, x = cnt;
if(l == r) return;
int mid = (l + r) / 2;
if(pos <= mid) update(l, mid, T[x].l, T[y].l, pos, k);
else update(mid + 1, r, T[x].r, T[y].r, pos, k);
}
int query(int l, int r, int x, int k){
if(l == r) return l;
int mid = (l + r) / 2, sum = T[T[x].l].sum;
if(sum <= k) return query(mid+1, r, T[x].r, k-sum);
else return query(l, mid, T[x].l, k);
}
int cal(int x){
int st = 1, ans = 0;
while(st != n+1){
ans++;
st = query(1, n+1, root[st], x);
}
return ans;
}
int main()
{
int t2, t1;
scanf("%d", &n);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
for(int i = n; i >= 1; i--){
if(pre[a[i]]!=0) update(1, n+1, t2, t1, pre[a[i]], -1), t1 = t2;
update(1, n+1, root[i], t1, i, 1);
t1 = root[i];
pre[a[i]] = i;
}
for(int i = 1; i <= n; i++){
printf("%d ", cal(i));
}
printf("\n");
return 0;
}
主席树在线一般可以转化成离线,这道题我们还可以用离线BIT来搞,由于这套题补的时间比较久了,我就没搞了,大概就是这样,下面是别人的代码: