题意:两种操作:
1 区间【l,r】变为一个值
2 查询 ,k<=5
解题:线段树每个点记录a[l]*l^k,统计区间和,直接查询即可。然后把要查询的值多项式展开转化一下,当k=5时,就要同时查询k=0,1,2,3,4,5的值。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string.h>
#include<algorithm>
using namespace std;
#define mod 1000000007
#define ls rt<<1
#define rs rt<<1|1
#define ll __int64
using namespace std;
const int maxn=200010;
struct T
{
int l,r;
ll flag;
ll sum[6];
}T[maxn<<2];
int a[maxn];
ll s2[maxn][6];
ll s1[maxn][6]= {0};
ll C[6][6];
ll s3[maxn][6];
void init()
{
int maxx=100010;
for(int i=0; i<6; i++)
{
C[i][0]=C[i][i]=1;
for(int j=1; j<i; j++)
C[i][j]=(C[i-1][j]+C[i-1][j-1]);
}
for(int i=1; i<=maxx; i++)
{
s2[i][0]=1;
s1[i][0]=i;
s3[i][0]=1;
}
for(int i=1; i<6; i++)
{
for(int j=1; j<=maxx; j++)
{
s2[j][i]=(s2[j][i-1]*j)%mod;
s1[j][i]=(s1[j-1][i]+s2[j][i])%mod;
s3[j][i]=((s3[j][i-1]*(1-j))%mod+mod)%mod;
}
}
}
void push_up(int rt)
{
// //0
// T[rt].sum[0]=T[ls].sum[0]+T[rs].sum[0];
// int len=T[ls].r-T[ls].l+1;
// //1
// T[rt].sum[1]=T[ls].sum[1]+T[rs].sum[1]+len*T[rs].sum[0];
// //2
// T[rt].sum[2]=T[ls].sum[2]+T[rs].sum[2]+len*len*T[rs].sum[0]+2*len*T[rs].sum[1];
// //3
// T[rt].sum[3]=T[ls].sum[3]+T[rs].sum[3]+len*len*len*T[rs].sum[0]+3*len*len*T[rs].sum[1]+3*len*T[rs].sum[2];
// //4
// T[rt].sum[4]=T[ls].sum[4]+T[rs].sum[4]+len*len*len*len*T[rs].sum[0]+6*len*len*T[rs].sum[2]+4*len*T[rs].sum[3]+4*len*len*len*T[rs].sum[1];
// //5
// T[rt].sum[5]=T[ls].sum[5]+T[rs].sum[5]+len*len*len*len*len*T[rs].sum[0]+5*len*len*len*len*T[rs].sum[1]+10*len*len*len*T[rs].sum[2]+10*len*len*T[rs].sum[3]+5*len*T[rs].sum[4];
for(int i=0; i<6; i++)
T[rt].sum[i]=(T[ls].sum[i]+T[rs].sum[i])%mod;
}
//void change(int rt,int val)
//{
// int l=T[rt].l;
// int r=T[rt].r;
// for(int i=0;i<=5;i++)
// {
// ll tmp=((s1[r][i]-s1[l-1][i])%mod+mod)%mod;
// T[rt].sum[i]=((ll)tmp*val)%mod;
// }
//}
void update(int rt,int left,int right,int x);
void push_down(int rt)
{
if(T[rt].flag!=-1)
{
int l=T[rt].l,r=T[rt].r,mid=(l+r)>>1;
update(ls,l,mid,T[rt].flag);
update(rs,mid+1,r,T[rt].flag);
T[rt].flag=-1;
}
}
void bulid(int rt,int left,int right)
{
T[rt].l=left;
T[rt].r=right;
T[rt].flag=-1;
if(left==right)
{
for(int i=0; i<6; i++)
T[rt].sum[i]=(a[left]*s2[left][i])%mod;
return ;
}
int mid=(left+right)>>1;
bulid(ls,left,mid);
bulid(rs,mid+1,right);
push_up(rt);
}
void update(int rt,int left,int right,int val)
{
if(T[rt].l==left&&T[rt].r==right)
{
T[rt].flag=val;
int l=T[rt].l;
int r=T[rt].r;
for(int i=0; i<6; i++)
{
ll tmp=((s1[r][i]-s1[l-1][i])%mod+mod)%mod;
T[rt].sum[i]=(tmp*val)%mod;
}
return ;
}
push_down(rt);
int mid=(T[rt].l+T[rt].r)>>1;
if(right<=mid) update(ls,left,right,val);
else if(left>mid) update(rs,left,right,val);
else
{
update(ls,left,mid,val);
update(rs,mid+1,right,val);
}
push_up(rt);
}
ll query(int rt,int left,int right,int k)
{
if(T[rt].l==left&&T[rt].r==right)
return T[rt].sum[k];
int mid=(T[rt].l+T[rt].r)>>1;
push_down(rt);
if(right<=mid) return query(ls,left,right,k);
else if(left>mid) return query(rs,left,right,k);
return (query(ls,left,mid,k)+query(rs,mid+1,right,k))%mod;
}
char op[10];
int _a,b,c;
int main()
{
int n,m;
// freopen("in.txt","r",stdin);
init();
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
bulid(1,1,n);
for(int i=1;i<=m;i++)
{
scanf("%s",&op);
scanf("%d%d%d",&_a,&b,&c);
if(op[0]=='=') update(1,_a,b,c);
else
{
ll ans=0;
for(int i=0; i<=c; i++)
ans=((((query(1,_a,b,i)*C[c][i])%mod*s3[_a][c-i])%mod+ans)%mod+mod)%mod;
printf("%I64d\n",ans);
}
}
}
return 0;
}