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;
}