牛牛的Link Power II(数组数组)

牛牛的Link Power II

http://www.nowcoder.com/questionTerminal/fd1203669c474535a45659842f2a1ffd

@[TOC]

题目链接:牛牛的Link Power II

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
牛牛有一颗大小为n的神奇Link-Cut 数组,数组上的每一个节点都有两种状态,一种为link状态,另一种为cut状态。数组上任意一对处于link状态的无序点对(即(u,v)和(v,u)被认为是同一对)会产生dis(u,v)的link能量,dis(u,v)为数组上u到v的距离。

我们定义整个数组的Link能量为所有处于link状态的节点产生的link能量之和。

一开始数组上每个节点的状态将由一个长度大小为n的01串给出,'1' 表示Link状态,'0' 表示Cut状态。

牛牛想要知道一开始,以及每次操作之后整个数组的Link能量,为了避免这个数字过于庞大,你只用输出答案对10^9^+7取余后的结果即可。

输入描述:
第一行输入一个正整数 n (1≤n≤10^5^)
接下里一行输入一个长度大小为n的01串表示数组的初始状态,'1'表示Link状态,'0'表示Cut状态。
接下来一行输入一个正整数m(1≤m≤10^5^) 表示操作的数目
接下来m行,每行输入两个正整数q,pos(q∈{1,2},1≤pos≤n)
当q=1时表示牛牛对数组的第pos个元素进行操作,将其赋值为1,保证在这个操作之前,该元素的值为0。
当q=2时表示牛牛对数组的第pos个元素进行操作,将其赋值为0,保证在这个操作之前,该元素的值为1。
输出描述:
请输出m+1行表示一开始,以及每次操作之后整个数组的Link能量,为了避免这个数字过于庞大,你只用输出答案对10^9^+7取余后的结果即可。
示例1
输入

5
00001
7
1 1
2 5
2 1
1 2
1 4
1 3
1 1

输出

0
4
0
0
0
2
4
10

题目大意

给你一个n,然后有一个长度为 n 的01字符串,问你任意两个1之间的距离之和;比如说111这三个 1 吧,输出结果就是 4,然后给你一个数 m 有m次操作每次操作有两个数 x,y。
x=1 就把位置为y的值变成1,然后输出所有的1的距离之和
x=2 就把位置为y的值变成0,然后输出所有的1的距离之和

解题思路

这个很明显的需要用线段树或树状数组去求解,这里我就用树状数组来讲解。我们很容易发如果给你一个序列,比如是 100101;这个01串,如果把第3个位置变成1,那么更变后多产生了多少值呢?产生了(1,3),(3,4),(3,6);这三段;如果把第5个位置变成1呢?那么多产生的就是(1,5),(4,5),(5,6);我们发现当我们插入值得时候,只会产生新的对数,对于之前的距离和不会改变,我们只需要吧之前的加上现在新增的就可以了。
如何求新增的呢?这里设插入位置为 y ,通过上面俩次模拟我么发现一个公式 新增加的数量就是y前面1的数量乘以y减去y前面的前缀和 再加上 y以后的后缀和减去 y乘以y后面1的数量;这里的后缀和就是位置唯一的值之和,比如说11011,前缀和sum5=1+2+4+5;这时候我们只需要维护一个前缀和就可以了。
!!!!!更重要的是记得最后加mod再次进行取模!!!!!不要问我为啥这么激动。。。。

代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int n;
const ll mod=1e9+7;
char b[100100];
ll sc[100100];// 存1的数量的前缀和 
ll sm[100100];// 存 值为1的坐标的前缀和 

void add(ll x,ll k)
{
    ll f=1;
    if(k<0) f=-1;//如果 k 小于 0的时候表示把1变成0需要减  
    for(ll i=x;i<=n;i+=(i&-i))
    {
        sc[i]+=f;
        sm[i]=(sm[i]+k)%mod;
    }
    return;
}
ll sum_c(ll x)//求前面 1 数量的前缀和 
{
    ll ans=0;
    for(ll i=x;i;i-=(i&-i))
    {
        ans=ans+sc[i];
    }
    return ans%mod;
}
ll sum_s(ll x)//求 值为1的坐标的前缀和
{
    ll ans=0;
    for(ll i=x;i;i-=(i&-i))
    {
        ans=(ans+sm[i])%mod;
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    scanf("%s",b+1);
    ll ans=0;
    for(ll i=1;i<=n;i++)
    {
        if(b[i]=='1')
        {
            add(i,i);//更新两个前缀和 
            //根据公式设刚开始是一个空的子串开始加 
            ans=(ans+sum_c(i-1)*i-sum_s(i-1))%mod;
            //因为刚开始是从前到后的所以不用管 i 后面的位置 
        }
    }
    cout<<ans<<"\n";
    ll m;
    cin>>m;
    ll x,y;
    while(m--)
    {
        cin>>x>>y;
        if(x==1)
        {
            ll sum_p=sum_s(y);//求出前面值为1的坐标的前缀合 
            ll sum_n=sum_s(n)-sum_p;//求后面值为1的坐标的前缀和 

            ll suc_p=sum_c(y);// 求前面 1 一共有多少个1 
            ll suc_n=sum_c(n)-suc_p;// 求后面一共有多少个1 
            //根据推出来的公式操作 
            ans=(ans+y*suc_p%mod-sum_p+sum_n-suc_n*y%mod)%mod;
            ans=(ans+mod)%mod;//这里一定记得再次取模!!!!!! 
            add(y,y);//更新 
            cout<<ans<<"\n";
        }
        else{
            add(y,-y);//这里先更新是为了更新后这个位置刚好是0;
                      //对求后面的值不影响     
            ll sum_p=sum_s(y);
            ll sum_n=sum_s(n)-sum_p;

            ll suc_p=sum_c(y);
            ll suc_n=sum_c(n)-suc_p;//这是把1变成0所以要减 
            ans=(ans-(y*suc_p-sum_p+sum_n-suc_n*y)%mod+mod)%mod;
            ans=(ans+mod)%mod;
            cout<<ans<<"\n";
        }
    }
    return 0;
}
全部评论

相关推荐

咦哟,从去年八月份开始长跑,两处实习转正都失败了,风雨飘摇,终于拿到offer了更新一下面试记录:秋招:多部门反复面试然后挂掉然后复活,具体问了啥已经忘了,只是被反复煎炸,直至焦香😋春招:base北京抖音hr打来电话说再次复活,准备面试,gogogo北京抖音一面:六道笔试题:1.promise顺序2.定义域问题3.flat展开4.并发请求5.岛屿数量算法(力扣)深度,广度都写6.忘记了,好像也是算法,难度中等其他问题多是框架底层设计,实习项目重难点~~~秒过😇北京抖音二面:三道笔试题:(为什么只有三道是因为第三道没做出来,卡住了)1.中等难度算法(忘记啥题了,应该是个数组的)2.认识js的继承本质(手写继承模式,深入js的面相对象开发)3.手写vue的响应式(卡在了watch,导致挂掉)---后知后觉是我的注册副作用函数写得有问题,有点紧张了其他题目多是项目拷打,项目亮点,对实习项目的贡献~~~第二天,挂,but立马复活转战深圳客服当天约面深圳客服一面:六道笔试题,由于面过太多次字节,面试官叫我直接写,不用讲,快些写完😋,具体都是些继承,深拷贝(注意对数组对象分开处理,深层次对象,循环引用),加中等难度算法题~~~秒过深圳客服二面:口诉八股大战:大概囊括网络,浏览器渲染原理,动画优化,时间循环,任务队列等等(你能想到的简单八股通通拉出来鞭尸😋)算法题:笔试题6道:1:找出数组内重复的数,arr[0]-arr[n]内的数大小为[1-n],例如[1,2,2,3,3]返回[2,3],要求o(n),且不使用任何额外空间(做到了o(n),空间方面欠佳,给面试官说进入下一题,做不来了)2:原滋原味的继承(所以继承真滴很重要)3:力扣股票购买时机难度中等其他滴也忘记了,因为拿到offer后鼠鼠一下子就落地了,脑子自动过滤掉可能会攻击鼠鼠的记忆😷~~~秒过深圳客服三面:项目大战参与战斗的人员有:成员1:表单封装及其底层原理,使用成本的优化,声明式表单成员2:公司内部库生命周期管理成员3:第三方库和内部库冲突如何源码断点调试并打补丁解决成员4:埋点的艺术成员5:线上项目捷报频传如何查出内鬼成员6:大文件分片的风流趣事成员7:设计模式对对碰成员8:我构建hooks应对经理的新增的小需求的故事可能项目回答的比较流利,笔试题3道,都很简单,相信大家应该都可以手拿把掐😇~~~过过过无hr面后续煎熬等待几天直接hr打电话发offer了,希望大家也可以拿到自己心仪的offer
法力无边年:牛哇,你真是准备得充分,我对你没有嫉妒,都是实打实付出
查看19道真题和解析
点赞 评论 收藏
分享
评论
6
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务