ZOJ - 3981 Balloon Robot 2017CCPC秦皇岛A题

ZOJ - 3981 Balloon Robot 2017CCPC秦皇岛A题

https://zoj.pintia.cn/problem-sets/91827364500/problems/91827370121

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<unordered_map>
#include<set>

#pragma GCC optimize(2)
using namespace std;
#define ll long long
#define PII pair<int,int>
#define PLL pair<ll,ll>
#define fi first
#define se second
#define pb push_back
#define debug(a) cout << #a << " " << a << '\n';
const int N = 1e5 + 5;
const int M = 1e5 + 5;
const int INF = 0x3f3f3f3f;
const ll LLINF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9 + 7;

inline ll read();

ll n, m, t;
ll s[N];
ll p;
ll num[N];

void solve() {
    vector<PLL > v;
    n = read();
    m = read();
    p = read();
    for (int i = 1; i <= n; i++)s[i] = read();
    ll sum = 0;
    for (int i = 0; i < p; i++) {
        ll a, b;
        a = read();
        b = read();
        ll val = s[a] - b % m;
        if (val < 0)val += m;
        num[i] = val;
        sum += num[i];
    }
    sort(num, num + p);//将所有队伍不开心值排序就相当于按照等待时间排序
    ll ans = sum;          //所以可以通过枚举等待时间递推
    for (int i = 0; i < p; i++) {//将这个等待时间置为0 比这个时间短的那些都要多等一个循环(m-num[i])，比这个时间短的都要少等一个num[i]
        ans = min(ans, sum + i * (m - num[i]) - (p - i) * num[i]);
    }
    cout << ans << '\n';


}

int main() {
    //ios::sync_with_stdio(false);
    t = read();
    while (t--) {
        solve();
    }

    return 0;
}


inline ll read() {
    char ch = getchar();
    ll p = 1, data = 0;
    while (ch < '0' || ch > '9') {
        if (ch == '-')p = -1;
        ch = getchar();
    }
    while (ch >= '0' && ch <= '9') {
        data = data * 10 + (ch ^ 48);
        ch = getchar();
    }
    return p * data;
}
/*
题意：
第一行三个数字n, m, q表示有m个座位围成一个环，n个队伍，q次A题
接下来n个数表示n个队伍所在位置(1<=ai<=m)
再接下来q行，每行a, b表示第a个队伍在第b秒A了一道题
有一个只会每一秒顺时针移动一个位置的发气球机器人
只要当前队伍有题目已经A了就会给他对应数量的气球（当然每道题最多1个气球）
如果a队伍在b时刻A了一道题，并在c时刻才拿到气球，那么这个队伍就会积累c-b点不开心值
求一个机器人起始位置（一开始是第0秒）使得所有队伍最终不开心值之和最小

分析：
假设机器人就在位置1，可以O(n)求出所有人的不开心值，排个序
之后暴力枚举初始位置，每移动1个位置可以使得所有不开心值不为0的队伍不开心值-1，
不开心值为0的队伍不开心值变为m，因为排过序所以这个可以O(1)转移
复杂度O(m)
m太大但其实有些位置一定不可能是最优的，
所以理论上只用枚举最多q个位置即可

那么就假设从1开始，得到每个气球的等待数组num。
给b排个序，用重复的只要算一次。 让i这个点等于0
（即减了num[i]），那么i前面的点都加了m-num[i]。
i后面的点都减了num[i]。就可以遍历一次答案，取最小。
*/