张经理的员工

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 131072K,其他语言262144K
64bit IO Format: %lld
题目描述 张经理的公司的办公室长达100000米,从最左端开始每间隔1米都有一个工位(从第1米开始有工位),位于第i米的工位称为i号工位,且这些工位都在一条水平线上。他有n个员工,每个员工分别位于xi号工位上(不同员工可能位于同一个工位)。

现在张经理想把员工聚集在某两个工位上,他有q套方案(每套方案包含两个工位号,两个工位号可能相同),他想知道对于每套方案,所有员工都到达两个工位中的某一个所需走的最短路径之和是多少。输入描述:第一行输入两个正整数n, q

第二行输入n个正整数xi,分别代表第i个员工的工位

之后q行每行输入两个整数a,b,代表该套方案要求的两个集合位置

(1<=n,q,xi,a,b<=10^5)(1<=n,q,xi,a,b<=105)输出描述:对于每套方案,输出一个整数代表答案,每个答案独占一行。
示例1
输入
3 2
1 3 5
1 4
2 1
输出
2
4

——————————————————————————————————————————————
比赛的时候没写出来,果然还是太菜了
思路是前缀和加预处理距离,但是判断写的太过于复杂把自己绕进去了,没想到后缀和处理

思路:有AB两点,小于等于A的去A,大于B的去B,小于等于[(A+B)/2]的去A,大于的去B;
输入数据后,进行前缀后缀处理:
前缀

size_t sumx=0;
 int sumn=0;
 for(int i=0;i<nmax;++i)
 {
  if(key[i])
  {
   sumx+=key[i]*i;//小于等于i位置的员工座标和 
   sumn+=key[i];//位置小于等于i的人数 
  }
  x[i]=sumx;
  num[i]=sumn;
 }

后缀

sumx=0;
 sumn=0;
 for(int i=nmax-1;i>=0;--i)
 {
  X[i]=sumx;
  NUM[i]=sumn;
  if(key[i])
  {
   sumx+=key[i]*i;
   sumn+=key[i];
  } 
 }

处理完后

  size_t avb=(a+b)/2;
  sum+=num[a]*a-x[a];//小于a
  sum+=X[b]-NUM[b]*b;//大于b
  sum+=(NUM[avb]-NUM[b])*b-X[avb]+X[b];//中点右
  sum+=x[avb]-x[a]-(num[avb]-num[a])*a;//中点左

AC码:

#include<bits/stdc++.h>
#define nmax 100001
using namespace std;
size_t x[100001]={0};
size_t X[100001]={0};
int key[100001]={0};
int num[1000001]={0};
int NUM[100001]={0};
int main()
{
 int n,q;
 cin>>n>>q;
 for(int i=0;i<n;++i)
 {
  int tmp;
  cin>>tmp;
  key[tmp]++;
 }
 size_t sumx=0;
 int sumn=0;
 for(int i=0;i<nmax;++i)
 {
  if(key[i])
  {
   sumx+=key[i]*i;//小于等于i位置的员工座标和 
   sumn+=key[i];//位置小于等于i的人数 
  }
  x[i]=sumx;
  num[i]=sumn;
 }
 sumx=0;
 sumn=0;
 for(int i=nmax-1;i>=0;--i)
 {
  X[i]=sumx;
  NUM[i]=sumn;
  if(key[i])
  {
   sumx+=key[i]*i;
   sumn+=key[i];
  } 
 }
 
 while(q--)
 {
  size_t sum=0;
  size_t c,d;
  cin>>c>>d;
  size_t a,b;
  a=min(c,d);
  b=max(c,d);
  size_t avb=(a+b)/2;
  sum+=num[a]*a-x[a];
  sum+=X[b]-NUM[b]*b;
  sum+=(NUM[avb]-NUM[b])*b-X[avb]+X[b];
  sum+=x[avb]-x[a]-(num[avb]-num[a])*a;
  cout<<sum<<endl;
 }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章