题目链接：https://www.nowcoder.com/acm/contest/91/A
思路：贪心（写复杂了）
数据弱，还可以直接暴力
1 #include<iostream>
2 #include<string>
3 #include<algorithm>
4 #include<cmath>
5 #include<string.h>
6 #include<stdio.h>
7 #include<queue>
8 using namespace std;
9 int a[100500];
10 int b[100500];
11 struct Node{
12 int num;
13 int value;
14 bool operator < (const Node &a) const {
15 return num>a.num;//最小值优先
16 }
17 };
18 int main(){
19 int T;
20 cin>>T;
21 while(T--){
22 memset(a,0,sizeof(a));
23 memset(b,0,sizeof(b));
24 int n;
25 cin>>n;
26 for(int i=0;i<n;i++){
27 scanf("%d",a+i);
28 }
29 for(int i=0;i<n;i++){
30 scanf("%d",b+i);
31 }
32 priority_queue< Node > q1,q2;
33 for(int i=0;i<n;i++){
34 Node tmp;
35 int value=a[i]-b[i];
36 tmp.num=i;
37 tmp.value=value;
38 if(value==0){
39 continue;
40 }else if(value<0){
41 q1.push(tmp);
42 }else{
43 q2.push(tmp);
44 }
45 }
46 long long ans=0;
47 while(!q2.empty()){
48 Node tmp1=q1.top();
49 Node tmp2=q2.top();
50 int x,y;
51 x=tmp1.value;
52 y=tmp2.value;
53 if(x+y==0){
54 ans+=y*abs(tmp1.num-tmp2.num);
55 q1.pop();
56 q2.pop();
57 }else if(x+y<0){
58 ans+=y*abs(tmp1.num-tmp2.num);
59 q1.pop();
60 q2.pop();
61 tmp1.value=x+y;
62 q1.push(tmp1);
63 }else{
64 ans+=(-x)*abs(tmp1.num-tmp2.num);
65 q1.pop();
66 q2.pop();
67 tmp2.value=x+y;
68 q2.push(tmp2);
69 }
70 }
71 cout<<ans<<endl;
72 }
73 return 0;
74 }
1 /*暴力*/
2 #include<iostream>
3 #include<bits/stdc++.h>
4 using namespace std;
5 long long a[100500];
6 long long b[100500];
7 int main(){
8 long long T;
9 cin>>T;
10 while(T--){
11 long long n;
12 cin>>n;
13 long long cnt=0;
14 for(long long i=0;i<n;i++){
15 scanf("%lld",a+i);
16 }
17 for(long long i=0;i<n;i++){
18 scanf("%lld",b+i);
19 }
20 for(long long i=0;i<n;i++){
21 if(a[i]==b[i]){
22 continue;
23 }
24 else if(a[i]<b[i]){
25 for(long long j=i;j<n;j++){
26 if(a[i]==b[i]){
27 break;
28 }
29 if(a[j]>b[j]){
30 long long value=a[j]-b[j];
31 if(value==b[i]-a[i]){
32 cnt+=value*abs(i-j);
33 a[i]=b[i];
34 a[j]=b[j];
35 }else if(value > b[i]-a[i]){
36 cnt+=(b[i]-a[i])*abs(i-j);
37 a[j]=a[j]-(b[i]-a[i]);
38 a[i]=b[i];
39 }else{
40 cnt+=value*abs(i-j);
41 a[i]=a[i]+value;
42 a[j]=b[j];
43 }
44 }
45 }
46 }else{
47 for(long long j=i;j<n;j++){
48 if(a[i]==b[i]){
49 break;
50 }
51 if(a[j]<b[j]){
52 long long value=b[j]-a[j];
53 if(value>a[i]-b[i]){
54 cnt+=(a[i]-b[i])*abs(i-j);
55 a[j]=a[j]+(a[i]-b[i]);
56 a[i]=b[i];
57 }else if(value<a[i]-b[i]){
58 cnt+=value*abs(i-j);
59 a[j]=b[j];
60 a[i]=a[i]-value;
61 }else{
62 cnt+=value*abs(i-j);
63 a[j]=b[j];
64 a[i]=b[i];
65 }
66 }
67 }
68 }
69 }
70 cout<<cnt<<endl;
71 }
72 return 0;
73 }