【解题报告】dutoj1184_电影院里的对话（树状数组）

题目链接：http://acm.dlut.edu.cn/problem.php?id=1184

Problem D: 电影院里的对话

Time Limit: 2 Sec   Memory Limit: 128 MB
Submit: 224   Solved: 38
[ Submit][ Status][ Web Board]

Description

电影开场后，小Y和小Q惊奇的发现电影无比的无聊，比《小时代》还要无聊。

于是他俩决定聊天。奇葩情侣的聊天方式当然要奇葩。

小Q给定n个单词，每个单词长度不超过10

她会询问m次，每次给定一个区间[l,r]，问小Y此区间内有多少个不同的单词？

作为一个man，怎么能被萌妹子问倒，你说呢？

所以，众程序猿们，来帮他解决这个简单的问题吧。

Input

多组数据

每组数据第一行2个正整数，n，m表示有n个单词，m组询问。(n<=10^5,m<=10^4)

接下来n行，每行一个单词

接下来m行，每行2个整数，l,r（1<=l<=r<=n）

Output

对于每组询问输出一行，输出询问区间内有多少个不同的单词

Sample Input

7 3

master

light

is

hen

tai

master

light

1 7

2 7

3 7

Sample Output

555

HINT

思路：

对询问按右区间 r 进行升序排序。按输入次序从小到大枚举单词，将每个单词插入至树状数组的当前位置（+1），然后利用map判断该单词以前是否出现过，若出现过，则在其原来的位置上插入 -1，取消掉其对于后面数组元素的影响。然后将该单词与其当前的位置建立映射关系。判断当前所处理的位置是否等于当前所要查询的右区间，若相等则可以利用前缀和计算结果 ans = sum(r) - sum(l-1)。

#include <cstdio>
#include <algorithm>
#include <map>
#include <iostream>
#include <string>
#include <string.h>

using namespace std;

const int MAX=100010;

int n,m;
string str[MAX];//输入
int tree[MAX];//树状数组
int ans[10010];//输出结果

struct query{
    int l,r,id;
    bool operator<(const query q)const{
        return r < q.r;
    }
}que[10010];

void insert(int t,int v){//插入树状数组
    for(int i=t;i<=n;i+=i&(-i)){
        tree[i]+=v;
    }
}

int sum(int t){//查询前缀和
    int res=0;
    while(t){
        res+=tree[t];
        t-=t&(-t);
    }
    return res;
}

int main()
{
    //freopen("in.txt","r",stdin);

    while(~scanf("%d%d",&n,&m)){
        memset(tree,0,sizeof(tree));

        for(int i=1;i<=n;++i) cin>>str[i];//scanf("%s",str[i]);
        for(int i=0;i<m;++i) scanf("%d%d",&que[i].l,&que[i].r),que[i].id=i;

        sort(que,que+m);//按右区间进行升序排序

        map<string,int> Map;

        int t=0;

        for(int i=1;i<=n;++i){
            insert(i,1);//将单词插入i位置
            if(Map.count(str[i])!=0){//以前出现过
                insert(Map[str[i]],-1);//消除以前单词的影响
            }
            Map[str[i]]=i;//将单词映射为其最新的位置

            while(t<m && i==que[t].r){//如果当前处理到的位置等于查询的右区间
                ans[que[t].id]=sum(que[t].r)-sum(que[t].l-1);
                ++t;
            }
        }
        for(int i=0;i<m;++i) printf("%d\n",ans[i]);
    }
    return 0;
}