bzoj 2038: [2009国家集训队]小Z的袜子(hose)

2038: [2009国家集训队]小Z的袜子(hose)
Time Limit: 20 Sec Memory Limit: 259 MB
Submit: 7176 Solved: 3308
[Submit][Status][Discuss]
Description

作为一个生活散漫的人，小Z每天早上都要耗费很久从一堆五颜六色的袜子中找出一双来穿。终于有一天，小Z再也无法忍受这恼人的找袜子过程，于是他决定听天由命……
具体来说，小Z把这N只袜子从1到N编号，然后从编号L到R(L 尽管小Z并不在意两只袜子是不是完整的一双，甚至不在意两只袜子是否一左一右，他却很在意袜子的颜色，毕竟穿两只不同色的袜子会很尴尬。
你的任务便是告诉小Z，他有多大的概率抽到两只颜色相同的袜子。当然，小Z希望这个概率尽量高，所以他可能会询问多个(L,R)以方便自己选择。

Input

输入文件第一行包含两个正整数N和M。N为袜子的数量，M为小Z所提的询问的数量。接下来一行包含N个正整数Ci，其中Ci表示第i只袜子的颜色，相同的颜色用相同的数字表示。再接下来M行，每行两个正整数L，R表示一个询问。

Output

包含M行，对于每个询问在一行中输出分数A/B表示从该询问的区间[L,R]中随机抽出两只袜子颜色相同的概率。若该概率为0则输出0/1，否则输出的A/B必须为最简分数。（详见样例）

Sample Input

6 4

1 2 3 3 3 2

2 6

1 3

3 5

1 6

Sample Output

2/5

0/1

1/1

4/15

【样例解释】

询问1：共C(5,2)=10种可能，其中抽出两个2有1种可能，抽出两个3有3种可能，概率为(1+3)/10=4/10=2/5。

询问2：共C(3,2)=3种可能，无法抽到颜色相同的袜子，概率为0/3=0/1。

询问3：共C(3,2)=3种可能，均为抽出两个3，概率为3/3=1/1。

注：上述C(a, b)表示组合数，组合数C(a, b)等价于在a个不同的物品中选取b个的选取方案数。

【数据规模和约定】

30%的数据中 N,M ≤ 5000；

60%的数据中 N,M ≤ 25000；

100%的数据中 N,M ≤ 50000，1 ≤ L < R ≤ N，Ci ≤ N。

---

【分析】
第一次自己A了道莫队算法的题，虽然A之前WA了好几次。
题目大意：给n个数，m个询问：l~r 区间内随机取两个数，求出这两个数相等的概率。
做了两道莫队的题总结出了点套路。莫队算法基于分块的思想，针对的是多个区间询问，核心是排序，但只能离线处理。莫队的思路很好想：将区间分块→将询问以左端点排序后插入区间中→对于每一个块中的询问进行右端点排序→暴力扫。经过排序之后，虽然是暴力扫，但时间复杂度低。
对于本题来讲，分母很好算: fenmu=(l-r+1)*(l-r)。分子在进行暴力扫的时候不断更新即可。

此题有坑:50000*50000>int_max，有的变量要用 long long。

---

【代码】

//bzoj 2038 小Z的袜子 
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
int n,m,L,R,res;
long long zi,mu;
int a[50001],get[50001];
long long ans[50001][2];
struct node
{
    long long l,r,num;
}ask[50001];
inline bool comp_l(const node &x,const node &y) {return x.l<y.l;}
inline bool comp_r(const node &x,const node &y) {if(x.r==y.r) return x.l<y.l;return x.r<y.r;}
inline int gcd(long long e,long long r) {if(e%r==0) return r;return gcd(r,e%r);}
inline void yue(long long &x,long long &y)
{
    int tmp=gcd(y,x);
    x=x/tmp,y=y/tmp;
}
int main()
{
    int i,j,k;
    scanf("%d%d",&n,&m);
    fo(i,1,n) scanf("%d",&a[i]);
    fo(i,1,m)
    {
        scanf("%lld%lld",&ask[i].l,&ask[i].r);
        ask[i].num=i;
    }
    sort(ask+1,ask+m+1,comp_l);
    int x=sqrt(n);j=0;i=1;
    while(i<=m)
    {
        j++;
        int b=i;
        while(ask[i].l<=j*x && i<=m) i++;
        if(j==x) sort(ask+b,ask+m+1,comp_r);
        else sort(ask+b,ask+i,comp_r);
    }
    long long ll=ask[1].l,rr=ask[1].r;
    mu=(rr-ll)*(rr-ll+1);
    fo(i,ll,rr)
    {
        zi+=2*get[a[i]];
        get[a[i]]++;
    }
    ans[ask[1].num][0]=zi,ans[ask[1].num][1]=mu;
    fo(i,2,m)
    {
        mu=(ask[i].r-ask[i].l)*(ask[i].r-ask[i].l+1);
        if(ll<ask[i].l)
          fo(ll,ll,ask[i].l-1)
          {
              get[a[ll]]--;
              zi-=2*get[a[ll]];
          }
        else
          for(ll=ll-1;ll>=ask[i].l;ll--)
          {
              zi+=2*get[a[ll]];
              get[a[ll]]++;
          }
        ll=ask[i].l;
        if(rr<ask[i].r)
          fo(rr,rr+1,ask[i].r)
          {
              zi+=2*get[a[rr]];
              get[a[rr]]++;
          }
        else
          for(rr;rr>ask[i].r;rr--)
          {
              get[a[rr]]--;
              zi-=2*get[a[rr]];
          }
        rr=ask[i].r;
        ans[ask[i].num][0]=zi;
        ans[ask[i].num][1]=mu;
    }
    fo(i,1,m)
    {
        if(!ans[i][0])
        {
            printf("0/1\n");
            continue;
        }
        yue(ans[i][0],ans[i][1]);
        printf("%lld/%lld\n",ans[i][0],ans[i][1]);
    }
    return 0;
}