cogs1619. [HEOI2012]采花 x

1619. [HEOI2012]采花

★★☆   输入文件：1flower.in   输出文件：1flower.out   简单对比
时间限制：5 s   内存限制：128 MB

【题目描述】

萧薰儿是古国的公主，平时的一大爱好是采花。

今天天气晴朗，阳光明媚，公主清晨便去了皇宫中新建的花园采花。花园足够大，容纳了n朵花，花有c种颜色（用整数1-c表示），且花是排成一排的，以便于公主采花。公主每次采花后会统计采到的花的颜色数，颜色数越多她会越高兴！同时，她有一癖好，她不允许最后自己采到的花中，某一颜色的花只有一朵。为此，公主每采一朵花，要么此前已采到此颜色的花，要么有相当正确的直觉告诉她，她必能再次采到此颜色的花。由于时间关系，公主只能走过花园连续的一段进行采花，便让女仆福涵洁安排行程。福涵洁综合各种因素拟定了m个行程，然后一一向你询问公主能采到多少朵花（她知道你是编程高手，定能快速给出答案！），最后会选择令公主最高兴的行程（为了拿到更多奖金！）。

【输入格式】

 第一行四个空格隔开的整数n、c以及m。接下来一行n个空格隔开的整数，每个数在[1, c]间，第i个数表示第i朵花的颜色。接下来m行每行两个空格隔开的整数l和r（l ≤ r），表示女仆安排的行程为公主经过第l到第r朵花进行采花。

【输出格式】

 

共m行，每行一个整数，第i个数表示公主在女仆的第i个行程中能采到的花的颜色数。

【样例输入】

5  3 5
  1  2 2 3 1
  1  5
  1  2
  2  2
  2  3
  3  5
  

【样例输出】

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的花各一朵，公主不采。
  

【提示】

【数据范围】

对于100%的数据，1 ≤ n ≤    10^6，c ≤ n，m ≤10^6。

【来源】

 

【题目来源】

耒阳大世界（衡阳八中） OJ 2743

思路:

　　1）首先暴力只能够拿到20分

　　2）莫队（真心不太懂。。。还没写）

　　3）离线+树状数组维护

上代码:

1）暴力（20）代码

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
 
using namespace std;

const int M = 1e6 + 1;
int n,c,m,ans;
int color[M],v[M];

inline int reads()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0' || ch>'9')
    {if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0' && ch<='9')
    {x=10*x+ch-'0';ch=getchar();}
    return x*f;
}

void orz(int l,int r)
{
    ans=0;
    memset(v,0,sizeof(v));
    for(int j=l;j<=r;j++)
        v[color[j]]++;
    for(int j=1;j<=c;j++)
        if(v[j]>1)
            ans++;
    printf("%d
",ans);
}

int main()
{
    n=reads();c=reads();m=reads();
    for(int i=1;i<=n;i++)
        color[i]=reads();
    for(int i=1,l,r;i<=m;i++)
    {
        l=reads();r=reads();
        orz(l,r);
    }
    return 0;
}

 2）（伪）正解：

　　据说是莫队???

为什么我写莫队T了2点qwq

T2点代码qwq:

#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;

const int M = 1000100;
int n,c,m,size,curL,curR,answer;
int a[M],cnt[M],ans[M];

struct G {
    int l,r,id,block;
    bool operator < (const G &qwq)const
    {
        if(block==qwq.block) 
            return r < qwq.r;
        return block < qwq.block;
    }
}t[M];

inline int read(int &qwq)
{
    char ch=' ';int q=0,w=1;
    for(;(ch!='-')&&((ch<'0')||(ch>'9'));ch=getchar());
    if(ch=='-') w=-1,ch=getchar();
    for(;ch>='0' && ch<='9';ch=getchar())q=q*10+ch-48;
    qwq=q*w;
    return qwq;
}

inline void add(int pre)
{
    cnt[a[pre]]++;
    if(cnt[a[pre]]==2)
        answer++;
}

inline void change(int pre)
{
    cnt[a[pre]]--;
    if(cnt[a[pre]]==1)
        answer--;
}

inline void solve()
{
    curL=1,curR=0;
    for(int i=1;i<=m;++i)
    {
        while(curL<t[i].l)
            change(curL),curL++;
        while(curL>t[i].l)
            curL--,add(curL);
        while(curR<t[i].r)
            curR++,add(curR);
        while(curR>t[i].r)
            change(curR),curR--;
        ans[t[i].id]=answer;
    }
}

int main()
{
    freopen("1flower.in","r",stdin);
    freopen("1flower.out","w",stdout);
    read(n),read(c),read(m);
    size=sqrt(n);
    for(int i=1;i<=n;++i)
        read(a[i]);
    for(int i=1;i<=m;++i)
    {
        read(t[i].l),read(t[i].r);
        t[i].block=(t[i].l-1)/size+1;
        t[i].id=i;
    }
    sort(t+1,t+m+1);
    solve();
    for(int i=1;i<=m;++i)
        printf("%d
",ans[i]);
    return 0;
}

难道是我的读入优化写得太丑了???orz

 应该不会!!!

3）正解:

　　　　离线+树状数组维护

#include<iostream>
#include<cstdio>
#include<algorithm>
#define Lowbit(x) (x&(-x))
using namespace std;

const int M = 1000010;
int n;
int a[M],gs[M],c[M],next[M],last[M];
int ans[M];

struct node {
    int l,r,id;
    bool operator < (const node &qwq)const
    {
        return l<qwq.l;
    }
}q[M];

int read()
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0' || ch>'9')
    {if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0' && ch<='9')
    {x=x*10+(ch-'0');ch=getchar();}
    return x*f;
}

void Update(int k,int x)
{
    while(k<=n)
    {
        c[k]+=x;
        k+=Lowbit(k);
    }
}

int Sum(int k)
{
    int sum=0;
    while(k>0)
    {
        sum+=c[k];
        k-=Lowbit(k);
    }
    return sum;
}

int main()
{
    freopen("1flower.in","r",stdin);
    freopen("1flower.out","w",stdout);
    int c,m,i,L;
    n=read(),c=read(),m=read();
    for(i=1; i<=n; i++)
        a[i]=read();
    for(i=1; i<=m; i++)
        q[i].l=read(),q[i].r=read(),q[i].id=i;
    sort(q+1,q+m+1);
    for(i=1; i<=n; i++)
    {
        if(last[a[i]]!=0)
            next[last[a[i]]]=i;
        last[a[i]]=i;
    }
    for(i=1; i<=n; i++)
    {
        gs[a[i]]++;
        if(gs[a[i]]==2)
            Update(i,1);
    }
    L=1;
    for(i=1; i<=m; i++)
    {
        while(L<q[i].l)
        {
            if(next[L]!=0)
                Update(next[L],-1);
            if(next[next[L]]!=0)
                Update(next[next[L]],1);
            L++;
        }
        ans[q[i].id]=Sum(q[i].r)-Sum(q[i].l-1);
    }
    for(i=1; i<=m; i++)
        printf("%d
",ans[i]);
    return 0;
}