当前位置: 代码迷 >> 综合 >> CCPC 2020威海站 H. Message Bomb (差分前缀和)
  详细解决方案

CCPC 2020威海站 H. Message Bomb (差分前缀和)

热度:82   发布时间:2023-12-22 13:13:56.0

PTA重现赛入口

题意: 有 n个组(初始都为空组), m个学生, s个事件。每次事件输入t, x, y。

  • t == 1: 将学生x加入y组,保证之前不存在。
  • t == 2: 将学生x从y组中删除,保证之前存在。
  • t == 3: 学生x向y组中其他成员宣布一则事件。

试求每个学生接收到的事件数。
在这里插入图片描述
思路:

  • 直接利用前缀和与差分的思维处理组员数的变化带来的改变。
  • 数组 vg[i]表示第i组总共接收过的事件数,数组vs[i]表示第i个学生总共接收到的事件数,再利用m个set集合存 每个学生加入的队伍情况。
  • 当y组新增一个成员x时,在 vs[x]上减去第y组之前接收到的事件数(差分)。
  • 当y组删除一个成员x时,在 vs[x]上加上第y组目前接收到的事件数(前缀和)。
  • 当成员x向y组内其他成员传送事件时,整个组的事件数 vg[y] ++,并且 成员x自身的事件数 vs[x] --。
  • 最后再对每个学生加入的队伍事件数求个总和即可。

代码实现:

#include<bits/stdc++.h>
#define endl '\n'
#define null NULL
#define ll long long
//#define int long long
#define pii pair<int, int>
#define lowbit(x) (x &(-x))
#define ls(x) x<<1
#define rs(x) (x<<1+1)
#define me(ar) memset(ar, 0, sizeof ar)
#define mem(ar,num) memset(ar, num, sizeof ar)
#define rp(i, n) for(int i = 0, i < n; i ++)
#define rep(i, a, n) for(int i = a; i <= n; i ++)
#define pre(i, n, a) for(int i = n; i >= a; i --)
#define IOS ios::sync_with_stdio(0); cin.tie(0);cout.tie(0);
const int way[4][2] = {
    {
    1, 0}, {
    -1, 0}, {
    0, 1}, {
    0, -1}};
using namespace std;
const int  inf = 0x7fffffff;
const double PI = acos(-1.0);
const double eps = 1e-6;
const ll   mod = 1e9 + 7;
const int  N = 2e5 + 5;int n, m, s;
int vg[N], vs[N];
set<int> mp[M];signed main()
{
    IOS;cin >> n >> m >> s;while(s --){
    int t, x, y; cin >> t >> x >> y;if(t == 1){
    vs[x] -= vg[y];mp[x].insert(y);}else if(t == 2){
    vs[x] += vg[y];mp[x].erase(y);}else{
    vg[y] ++;vs[x] --;}}for(int i = 1; i <= m; i ++){
    for(auto it : mp[i]) vs[i] += vg[it];}for(int i = 1; i <= m; i ++)cout << vs[i] << endl;return 0;
}
  相关解决方案