CodeForces 145 D.Lucky Pair（组合数学+set）

Description

给出一个长度为$n$的序列，称一对区间$[L_1,R_1],[L_2,R_2]$是合法的，当且仅当这两个区间无交集且这两个区间中不出现相同的幸运数字，幸运数字即全部由$4$和$7$组成的数字，问满足条件的区间对数

Input

第一行输入一整数$n$，之后输入$n$个整数$a_i$$(2\le n\le 10^5,1\le a_i\le 10^9)$

Output

输出满足条件的区间对数

Sample Input

4
1 4 2 4

Sample Output

9

Solution

首先找到所有的幸运数字，假设共$m$个，位置为$pos_1,pos_2,...,pos_m$

从$n$个位置中取出两个不相交区间的方案数为$C_n^4+2\cdot C_n^3+C_n^2$，考虑求出不合法区间对数

枚举第二个区间包含第$i$个幸运数字到第$j$个幸运数字，预处理每个幸运数字前一个和其相同的幸运数字位置，固定$i$后，随着$j$的增大，第一个区间不能取的幸运数字越来越多，$1$~$pos_{i}-1$这段区间就会被分割成很多小区间，每一个小区间的子区间都可以作为合法的第一个区间，其他都不行，假设当前加入了第$j$个幸运数字，第$k$个幸运数字和第$j$个幸运数字相同并且$pos_k<pos_i$，说明之间的某一个合法区间被$pos_k$分割成了两个更小的合法区间，我们用$set$维护$1$~$pos_{i}-1$这个区间中不合法位置以便维护合法区间

假设我们找到两个相邻的不合法位置$L,R$满足$L<pos_k<R$，考虑此时产生的不合法答案数量，当第一个区间左端点介于$(L,pos_k]$之间，第一个左区间右端点介于$[pos_k,R)$之间时，第二个区间左端点介于$(pos_{i-1},pos_i]$时，第二个区间右端点介于$[pos_j,pos_{j+1})$之间时不合法，方案数$(pos_k-L)\cdot(R-pos_k)\cdot(pos_i-pos_{i-1})\cdot(pos_{j+1}-pos_j)$

假设$pos_k$是最大的不合法位置，设其前一个位置是$L$，那么产生的不合法方案即为，第一个区间左端点介于$(L,pos_k]$，第一个区间右端点介于$[pos_k,pos_{i-1})$，第二个区间左端点介于$(pos_{i-1},pos_i]$，第二个区间右端点介于$[pos_j,pos_{j+1})$之间，方案数$(pos_k-L)\cdot(pos_{i-1}-pos_k)\cdot(pos_i-pos_{i-1})\cdot(pos_{j+1}-pos_j)$，另一类不合法方案为，第一个区间左端点介于$(L,pos_k]$，第一个区间右端点和第二个区间左端点均介于$[pos_{i-1},pos_i]$之间，第二个区间右端点介于$[pos_j,pos_{j+1})$之间，方案数$(pos_k-L)\cdot C_{pos_i-pos_{i-1}+1}^{2}\cdot(pos_{j+1}-pos_j)$

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef unsigned long long ull;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1005;
int n,a[100005],b[maxn],pos[maxn],pre[maxn];
set<int>s;
set<int>::iterator l,r;
bool check(int n)
{
    while(n)
    {
        if(n%10!=4&&n%10!=7)return 0;
        n/=10;
    }
    return 1;
}
ull C(int n,int m)
{
    if(m==2)return (ull)n*(n-1)/2;
    if(m==3)return (ull)n*(n-1)*(n-2)/6;
    if(m==4)return (ull)n*(n-1)*(n-2)/6*(n-3)/4;
}
int main()
{
    while(~scanf("%d",&n))
    {
        int m=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            if(check(a[i]))pos[++m]=i;
        }
        pos[m+1]=n+1;
        for(int i=1;i<=m;i++)
        {
            pre[i]=0;
            for(int j=i-1;j>=1;j--)
                if(a[pos[i]]==a[pos[j]])
                {
                    pre[i]=j;
                    break;
                }
        }
        ull ans=C(n,4)+2ull*C(n,3)+C(n,2);
        for(int i=2;i<=m;i++)
        {
            s.clear();
            s.insert(0);
            ull res=0;
            for(int j=i;j<=m;j++)
            {
                if(pre[j]<i)
                {
                    for(int k=pre[j];k;k=pre[k])
                    {
                        l=r=s.lower_bound(pos[k]);
                        l--;
                        if(r==s.end())
                        {
                            res+=(ull)(pos[k]-*l)*(pos[i-1]-pos[k])*(pos[i]-pos[i-1]);
                            res+=(ull)(pos[k]-*l)*C(pos[i]-pos[i-1]+1,2);
                        }
                        else res+=(ull)(pos[k]-*l)*(*r-pos[k])*(pos[i]-pos[i-1]);
                        s.insert(pos[k]);
                    }
                }
                ans-=res*(pos[j+1]-pos[j]);
            }
        }
        printf("%I64u\n",ans);
    }
    return 0;
}