中位数定理

Arranging The Sheep
中位数定理:所有数到中间位置之和最短

#include<stdio.h>
#include<string.h>
#include<queue>
#include<math.h>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int N=1e6+10;
#define inf 0x3f3f3f3f3f
char a[N];
priority_queue<ll>q;
//不停向中间靠近 存一下每个羊有多少个草垛 找到每个样距离中间的位置之和
ll book[N];
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
       // memset(book,0,sizeof(book));
        int n,k=0;
        ll sum=0,s=0;
        scanf("%d",&n);
        scanf("%s",a);
        int l=strlen(a);
        for(int i=0; i<l; i++)
        {
            if(a[i]=='.')
            {
                s++;
            }
            else
                book[k++]=s;   //记录每个羊前面的草垛个数
        }
        //for(int i=0;i<k;i++)
        //    printf("@@@%d\n",book[i]);
        int m=k/2;
        for(int i=0; i<k; i++)
        {
            sum+=abs(book[m]-book[i]);
        }
        printf("%lld\n",sum);
    }
    return 0;
}