牛客网每日刷题(3)

阅读数: 次 2021-01-30

1. 4 Values Whose Sum is Zero

1.1. 题目

链接：https://vjudge.z180.cn/problem/UVA-11054
来源：vjudge

As you may know from the comic “Asterix and the Chieftain’s Shield”, Gergovia consists of one street,
and every inhabitant of the city is a wine salesman. You wonder how this economy works? Simple
enough: everyone buys wine from other inhabitants of the city. Every day each inhabitant decides how
much wine he wants to buy or sell. Interestingly, demand and supply is always the same, so that each
inhabitant gets what he wants.
There is one problem, however: Transporting wine from one house to another results in work. Since
all wines are equally good, the inhabitants of Gergovia don’t care which persons they are doing trade
with, they are only interested in selling or buying a specific amount of wine. They are clever enough
to figure out a way of trading so that the overall amount of work needed for transports is minimized.
In this problem you are asked to reconstruct the trading during one day in Gergovia. For simplicity
we will assume that the houses are built along a straight line with equal distance between adjacent
houses. Transporting one bottle of wine from one house to an adjacent house results in one unit of
work.
Input
The input consists of several test cases. Each test case starts with the number of inhabitants n
(2 ≤ n ≤ 100000). The following line contains n integers ai (−1000 ≤ ai ≤ 1000). If ai ≥ 0, it
means that the inhabitant living in the i-th house wants to buy ai bottles of wine, otherwise if ai < 0,
he wants to sell −ai bottles of wine. You may assume that the numbers ai sum up to 0.
The last test case is followed by a line containing ‘0’.
Output
For each test case print the minimum amount of work units needed so that every inhabitant has his
demand fulfilled. You may assume that this number fits into a signed 64-bit integer (in C/C++ you
can use the data type “long long”, in JAVA the data type “long”).

1.2. 样例输入&&样例输出

5
5 -4 1 -3 1
6
-1000 -1000 -1000 1000 1000 1000
0

1
2

9
9000

1.3. 思路

把a+b+c+d=0的问题转化成a+b=-(c+d)问题
写一个map然后加起来即可

1.4. AC代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<vector>
#include<sstream>
#include<map>
#include<unordered_map>
using namespace std;
const int maxn = 1000;
const int maxnn = 100000;
typedef long long ll;
int a[maxnn],b[maxnn],c[maxnn],d[maxnn];
int main() {
    int t;cin>>t;
    int kase=0;
    while(t--){
        if(kase==0)kase++;else cout<<endl;
        int n;cin>>n;
        for(int i=0;i<n;i++){
            cin>>a[i];
            cin>>b[i];
            cin>>c[i];
            cin>>d[i];
        }
        unordered_map<int,int>mp;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                mp[a[i]+b[j]]++;
            }
        }
        int cnt=0;
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                cnt+=mp[-(c[i]+d[j])];
            }
        }
        cout<<cnt<<endl;
    }
    system("pause");
    return 0;
}