AtCoder Beginner Contest 171 D Replacing 桶排序+及時更新數據

AtCoder Beginner Contest 171   比賽人數10532  比賽開始後14分鐘看到A題,之後,每過一分鐘看到一道題

AtCoder Beginner Contest 171  D   Replacing    桶排序+及時更新數據

總目錄詳見https://blog.csdn.net/mrcrack/article/details/104454762

在線測評地址https://atcoder.jp/contests/abc171/tasks/abc171_d

題目大意：給定一個數組，在Q次更新中，每進行一次更新時，將數組元素值爲B的元素，值全更新爲C,輸出每次更新後，數值中的元素和。

基本思路：桶排序+及時更新。爲了更好的理解代碼，代碼中的測試語句只是註釋掉。

AC代碼如下：

#include <stdio.h>
#define LL long long
#define maxn 100010
LL sum,a[maxn],cnt[maxn];
int main(){
	int i,n,q,b,c;
	scanf("%d",&n);
	for(i=1;i<=n;i++)scanf("%lld",&a[i]),sum+=a[i],cnt[a[i]]++;
	//printf("sum=%lld\n",sum);
	scanf("%d",&q);
	while(q--){
		scanf("%d%d",&b,&c);
		//printf("cnt[%d]=%lld cnt[%d]=%lld\n",b,cnt[b],c,cnt[c]);
		sum-=cnt[b]*b;//刪除舊數據
		//printf("sum=%lld\n",sum);
		sum+=cnt[b]*c;//加上新數據
		//printf("sum=%lld\n",sum);
		cnt[c]+=cnt[b],cnt[b]=0;//更新新數據數量
		printf("%lld\n",sum);
	}
	return 0;
}