Spaceship Solitaire【Codeforces 1266 E】【思维】

Codeforces Global Round 6 E

---

  题意：主人公想要造宇宙飞船，所以需要N种物资，每种物资的需求量是a[i]个。然后呢，如果我们没有任何加速器的话，总的时间需求是，但是现在我们有“里程碑”加速器！

  里程碑加速器是这样的「S，T，U」我们如果有T个S物资的话，我们可以免费获得一个U物资。

  思路：那么，不难发现，如果U这个物资的数量不足a[U]，那么就是可以减少一次步骤的，如果大于了，就是没有作用的了，然后map或者unordered_map维护一下即可。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <unordered_map>
#include <unordered_set>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f3f3f3f3f
#define HalF (l + r)>>1
#define lsn rt<<1
#define rsn rt<<1|1
#define Lson lsn, l, mid
#define Rson rsn, mid+1, r
#define QL Lson, ql, qr
#define QR Rson, ql, qr
#define myself rt, l, r
#define MP(a, b) make_pair(a, b)
using namespace std;
typedef unsigned long long ull;
typedef unsigned int uit;
typedef long long ll;
const int maxN = 2e5 + 7;
int N, Q;
ll a[maxN], update[maxN];
struct hashfunc
{
    template<typename T, typename U>
    size_t operator() (const pair<T, U> &i) const
    {
        return hash<T>()(i.first) ^ hash<U>()(i.second);
    }
};
unordered_map< pair<int, int>, int , hashfunc > mp;
pair<int, int> pir;
int main()
{
    ll ans = 0;
    scanf("%d", &N);
    for(int i=1; i<=N; i++)
    {
        scanf("%lld", &a[i]);
        ans += a[i];
    }
    scanf("%d", &Q);
    int S, T, U, las_U;
    while(Q--)
    {
        scanf("%d%d%d", &S, &T, &U);
        pir = MP(S, T);
        if(!mp[pir])
        {
            mp[pir] = U;
            update[U] ++;
            if(update[U] <= a[U])
            {
                ans --;
            }
        }
        else
        {
            las_U = mp[pir];
            update[las_U] --;
            if(update[las_U] < a[las_U])
            {
                ans ++;
            }
            mp[pir] = U;
            if(U)
            {
                update[U] ++;
                if(update[U] <= a[U])
                {
                    ans --;
                }
            }
        }
        printf("%lld\n", ans);
    }
    return 0;
}