power

第一行为一个正整数 $n$ ，指明以下的数组长度为 $n$ 。

第二行 $01$ 串

第三行为一个正整数 $Q$ ，指明以下的查询数为 $Q$ 。

第三行到第 $Q+2$ 行表示 $Q$ 个查询

每个查询输入一个整数 $1 \le id\le n$

要求计算 $01$ 串中，所有为 $1$ 的索引到 $id$ 的距离和

>索引 $i$ 到 $id$ 的距离为 $abs(id-i)$
>
>索引从 $1$ 开始

C++ 解法, 执行用时: 6ms, 内存消耗: 460K, 提交时间: 2021-12-14 22:16:50

#include<bits/stdc++.h>
using namespace std;
int main(){
	int n,q,zf=0,sum=0;
	string a;
	cin>>n;
    cin>>a;
	cin>>q;
	while(q--){
		sum=0;
		cin>>zf;
		zf--;
		for(int i=0;i<n;i++){
			if(a[i]=='1') {
			sum+=abs(zf-i);}
		}
		cout<<sum<<endl;
	}
}

pypy3 解法, 执行用时: 159ms, 内存消耗: 26864K, 提交时间: 2021-12-12 14:23:37

n=int(input())
zo=str(input())

T=int(input())

for i in range(T):
    idn=int(input())
    m=0
    t=0
    for i in zo:
        t+=1
        if i=='1':
            m+=abs(idn-t)
    print(m)

Python3 解法, 执行用时: 242ms, 内存消耗: 4580K, 提交时间: 2022-04-28 16:56:43

ll=int(input())+1
l='0'+input()
for _ in range(int(input())):
    ans=0
    n=int(input())
    for x in range(1,ll):
        if l[x]=='1':
            ans+=abs(n-x)
    print(ans)

详情