首页 > 代码库 > {CodeForces】788E New task && 汕头市队赛SRM06 D 五色战队
{CodeForces】788E New task && 汕头市队赛SRM06 D 五色战队
D 五色战队 SRM 06
背景&&描述
反正就是b,c,d这三人一定要有头。
输入格式
第一行一个整数n,第二行n个整数表示人们的发色。
第三行一个整数m,表示操作数。
接下来有m行,每行第一个数表示操作类型,第二个数表示被操作那人的编号。
类型为1就是把他变无头,类型2就是变有头。
输出格式
每次操作后输出答案,答案对10^9+7取模。。。
样例输入
8 3 4 4 2 4 5 4 1 3 1 5 2 5 1 2
样例输出
1 6 2
数据范围与约定
- 对于100%的数据:
样例解释
第一次操作后可能的战队为{1,2,3,7,8}
第二次操作后可能的战队为{1,2,3,5,7},{1,2,3,5,8},{1,2,3,7,8},{1,2,5,7,8},{1,3,5,7,8},{2,3,5,7,8}。
第三次操作后可能的战队为{1,3,5,7,8},{2,3,5,7,8}
这道题和codechef的原题差不多
题目要求的就是
n个数字中,每个数有数字A和属性B,每次操作将某个点x的属性B改变为0或1,求满足这样要求的子序列的个数:
下标a<b<c<d<e,而Aa<=Ab=Ac=Ad>=Ae且Bb=Bc=Bd=1。
那么每个数左边以及右边小于等于他的数 很显然我们可以用树状数组维护 下标就是值
当然为了方便我们需要将值离散化一波
这样之后 重点就是中间那三个数了 我们可以按值来搞 每个值建一棵线段树
需要维护的信息有(我们设五个点分别为 1 2 3 4 5
sz 及点的个数
A 就是 1 2
C 就是 2 3
AB就是 1 2 3
BC 就是 3 4 5
ABC 就是 1 2 3 4 5
具体怎么算看up咯
跟结点只需要算A C 而A C可以由左边比他小和右边比他小的点的数量得到 这样之后就简单多了 2333
#include<cstdio> #include<cstring> #include<algorithm> #define LL long long using namespace std; const int M=2000007,N=200007,mod=1e9+7; int read(){ int ans=0,f=1,c=getchar(); while(c<‘0‘||c>‘9‘){if(c==‘-‘) f=-1; c=getchar();} while(c>=‘0‘&&c<=‘9‘){ans=ans*10+(c-‘0‘); c=getchar();} return ans*f; } int sum[N],s[N],ans; struct node{int w,pos;}e[N]; bool cmp(node a,node b){return a.w<b.w;} int n,m,cnt,sz; int L[N],R[N],root[N]; void clear(){memset(sum,0,sizeof(sum));} int lowbit(int x){return x&-x;} void add(int x){ while(x<=n){ sum[x]++; x+=lowbit(x); } } int push_sum(int x){ int ans=0; while(x){ans+=sum[x]; x-=lowbit(x);} return ans; } struct note{ int l,r,sz; int A,C,AB,BC,ABC; }tr[M]; void up(int x){ int l=tr[x].l,r=tr[x].r; tr[x].sz=(tr[l].sz+tr[r].sz)%mod; tr[x].A=(tr[l].A+tr[r].A)%mod; tr[x].C=(tr[l].C+tr[r].C)%mod; tr[x].AB=(tr[l].AB+tr[r].AB+(LL)tr[l].A*tr[r].sz)%mod; tr[x].BC=(tr[l].BC+tr[r].BC+(LL)tr[r].C*tr[l].sz)%mod; tr[x].ABC=(tr[l].ABC+tr[r].ABC+(LL)tr[l].A*tr[r].BC+(LL)tr[l].AB*tr[r].C)%mod; } void mdf(int& x,int l,int r,int k,int v){ if(!x) x=++sz; if(l==r){ tr[x].sz=1*v; tr[x].A=L[k]*v; tr[x].C=R[k]*v; tr[x].AB=tr[x].BC=tr[x].ABC=0; tr[x].l=tr[x].r=0; return ; } int mid=(l+r)>>1; if(k<=mid) mdf(tr[x].l,l,mid,k,v); else mdf(tr[x].r,mid+1,r,k,v); up(x); } int main() { n=read(); for(int i=1;i<=n;i++) e[i].w=read(),e[i].pos=i; sort(e+1,e+1+n,cmp); for(int i=1;i<=n;i++){ if(e[i].w!=e[i-1].w) cnt++; s[e[i].pos]=cnt; } clear(); for(int i=1;i<=n;i++){ L[i]=push_sum(s[i]); add(s[i]); } clear(); for(int i=n;i>=1;i--){ R[i]=push_sum(s[i]); add(s[i]); } for(int i=1;i<=n;i++){ ans=(ans-tr[root[s[i]]].ABC+mod)%mod; mdf(root[s[i]],1,n,i,1); ans=(ans+tr[root[s[i]]].ABC)%mod; } int x,y; m=read(); for(int i=1;i<=m;i++){ x=read(); y=read(); ans=(ans-tr[root[s[y]]].ABC+mod)%mod; mdf(root[s[y]],1,n,y,x-1); ans=(ans+tr[root[s[y]]].ABC)%mod; printf("%d\n",ans); } return 0; }
{CodeForces】788E New task && 汕头市队赛SRM06 D 五色战队