[BZOJ2743][HEOI2012][樹狀數組]採花

Description
蕭芸斕是Z國的公主,平時的一大愛好是採花。
今天天氣晴朗,陽光明媚,公主清晨便去了皇宮中新建的花園採花。花園足夠大,容納了n朵花,花有c種顏色(用整數1-c表示),且花是排成一排的,以便於公主採花。公主每次採花後會統計採到的花的顏色數,顏色數越多她會越高興!同時,她有一癖好,她不允許最後自己採到的花中,某一顏色的花只有一朵。爲此,公主每採一朵花,要麼此前已採到此顏色的花,要麼有相當正確的直覺告訴她,她必能再次採到此顏色的花。由於時間關係,公主只能走過花園連續的一段進行採花,便讓女僕福涵潔安排行程。福涵潔綜合各種因素擬定了m個行程,然後一一向你詢問公主能採到多少朵花(她知道你是編程高手,定能快速給出答案!),最後會選擇令公主最高興的行程(爲了拿到更多獎金!)。
Input
第一行四個空格隔開的整數n、c以及m。接下來一行n個空格隔開的整數,每個數在[1, c]間,第i個數表示第i朵花的顏色。接下來m行每行兩個空格隔開的整數l和r(l ≤ r),表示女僕安排的行程爲公主經過第l到第r朵花進行採花。
Output
共m行,每行一個整數,第i個數表示公主在女僕的第i個行程中能採到的花的顏色數。
Sample Input
5 3 5
1 2 2 3 1
1 5
1 2
2 2
2 3
3 5
Sample Output
2
0 0 1 0
【樣例說明】
詢問[1, 5]:公主採顏色爲1和2的花,由於顏色3的花只有一朵,公主不採;詢問[1, 2]:顏色1和顏色2的花均只有一朵,公主不採;
詢問[2, 2]:顏色2的花只有一朵,公主不採;
詢問[2, 3]:由於顏色2的花有兩朵,公主採顏色2的花;
詢問[3, 5]:顏色1、2、3的花各一朵,公主不採。
HINT
【數據範圍】
對於100%的數據,1 ≤ n ≤ 10^6,c ≤ n,m ≤10^6。

同SDOI那道題,只是這裏多了一個至少採兩個顏色相同花的限制。我們就可以一開始在head[col]後繼的後繼sum++, 然後從左往右掃時 需要將當前pos的後繼sum–,因爲跳出pos後下一朵花也不可以採了

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e6+5;
int n,c,m,bit[maxn],col[maxn],head[maxn],nxt[maxn];
struct Route
{
    int L,R,ID,ans;
}route[maxn];
#define L(x) route[x].L
#define R(x) route[x].R
#define ID(x) route[x].ID
#define ans(x) route[x].ans
bool cmp1(Route x,Route y)
{
    return x.L<y.L;
}
bool cmp2(Route x,Route y)
{
    return x.ID<y.ID;
}
#define lowbit(x) x&(-x)
void update(int x,int val)
{
    for (int i=x;i<=n;i+=lowbit(i)) bit[i]+=val;
}
int query(int x)
{
    int ret=0;
    for (int i=x;i>0;i-=lowbit(i)) ret+=bit[i];
    return ret;
}
int main()
{
    scanf("%d%d%d",&n,&c,&m);
    for (int i=1;i<=n;i++) scanf("%d",&col[i]);//color
    for (int i=n;i>0;i--) nxt[i]=head[col[i]],head[col[i]]=i;
    for (int i=1;i<=c;i++) if (nxt[head[i]]) update(nxt[head[i]],1);
    for (int i=1;i<=m;i++) scanf("%d%d",&L(i),&R(i)),ID(i)=i;
    sort(route+1,route+m+1,cmp1);
    int pos=0;
    for (int i=1;i<=m;i++)
    {
        while (pos<L(i))
        {
            if (nxt[pos]) update(nxt[pos],-1);
            if (nxt[nxt[pos]]) update(nxt[nxt[pos]],1);
            pos++;
        }
        ans(i)=query(R(i)-query(L(i)-1));
    }
    sort(route+1,route+m+1,cmp2);
    for (int i=1;i<=m;i++) printf("%d\n",ans(i));
    return 0;
} 
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章