C-氧气少年的水滴 2_牛客练习赛121 (nowcoder.com)
氧气少年将 n 个水滴排成一排,第 i 个水滴的饱和值为ai(1≤ai≤9)。
每个水滴可以吸收其他水滴。例如:某一水滴目前的饱和值为a,它吸收了一个饱和值为b 的水滴,那么此水滴的饱和值变为a+b。
当某个水滴的饱和值增加到 10 的时候,这个水滴就会爆裂为两个饱和值为 1 的水滴(原水滴消失),分别向左和向右飞去,接下来:
- 对于向左飞的水滴,他会飞到与原水滴左侧相邻的水滴处,被左侧相邻的水滴吸收(如果左侧有相邻的水滴),或飞到向左无穷远的位置。
- 对于向右飞的水滴,他会飞到与原水滴右侧相邻的水滴处,被右侧相邻的水滴吸收(如果右侧有相邻的水滴),或飞到向右无穷远的位置。
请注意,对于空中两个相遇的、相向而行的水滴,它们只会"擦肩而过",换句话说,它们不受相遇水滴的任何影响。
已知氧气少年在第 p 个水滴上添加了一滴饱和值为 1 的水滴,请求出最终飞到向左无穷远的位置的水滴的数量和最终飞到向右无穷远的位置的水滴的数量。
输入描述:
第一行包含一个整数 T,表示测试用例的组数。 对于每组测试用例: 第一行包含两个整数 n,p(1≤p≤n),分别表示水滴的数量和添加水滴的位置。 第二行包含 n 个整数 a1…an(1≤ai≤9),表示水滴的饱和值。
输出描述:
对于每组测试用例: 仅输出一行,包含两个整数,分别表示最终飞到向左无穷远的位置的水滴的数量和最终飞到向右无穷远的位置的水滴的数量。
示例1
输入
复制3
3 3 2 9 9 9 4 2 9 9 9 8 3 2 9 8 9
输出
复制2
2 2 3 1 0 0
#include<iostream>
using namespace std;
int a[2100000];
int main (){
int t;
cin>>t;
int n,p;
int lp,rp;
int sumr,suml;
while(t--){
cin>>n>>p;
int b=0;
sumr=0,suml=0;
lp=p-2;
rp=p;
for(int i=0,x;i<n;i++){
cin>>x;
a[i]=x;
}
if(a[p-1]<9){
cout<<0<<" "<<0<<endl;
continue;
}
else
suml++,sumr++;
while(1){
b=0;
while(lp>=0&&a[lp]>=10-suml){
suml-=10-a[lp];
lp--;
sumr++,suml++;
b=1;
}
while(rp<=n-1&&a[rp]>=10-sumr){
sumr-=10-a[rp];
rp++;
sumr++,suml++;
b=1;
}
//最后进行判断水滴是否能够飞出来
if(b!=1){
if(lp!=-1)
suml=0;
if(rp!=n)
sumr=0;
break;}
}
cout<<suml<<" "<<sumr<<endl;
}
return 0;
}