洛谷 P3998 单链表解法

tqxy

已于 2024-01-21 13:37:49 修改

阅读量953

点赞数 25

文章标签：算法

于 2024-01-21 13:31:24 首次发布

本文链接：https://blog.csdn.net/qwc62668963/article/details/135728306

版权

洛谷 P3998 链接，点击此处

[SHOI2013] 发微博

题目描述

刚开通的 SH 微博共有 $n$ 个用户（ $1\sim n$ 标号），在这短短一个月的时间内，用户们活动频繁，共有 $m$ 条按时间顺序的记录：

! x 表示用户 x 发了一条微博；
+ x y 表示用户 x 和用户 y 成为了好友
− x y 表示用户 x 和用户 y 解除了好友关系

当一个用户发微博的时候，所有他的好友（直接关系）都会看到他的消息。

假设最开始所有人之间都不是好友关系，记录也都是合法的（即 + x y 时 $x$ 和 $y$ 一定不是好友，而 − x y 时 $x$ 和 $y$ 一定是好友）。

问这 $m$ 条记录发生之后，每个用户分别看到了多少条消息。

输入格式

第 $1$ 行两个整数 $n$ , $m$ 。

接下来 $m$ 行，按时间顺序读入 $m$ 条记录，每条记录的格式如题目所述，用空格隔开。

输出格式

输出一行 $n$ 个用空格隔开的数（行末无空格），第 $i$ 个数表示用户 $i$ 最后看到了几条消息。

样例 #1

样例输入 #1

2 8
! 1
! 2
+ 1 2
! 1
! 2
- 1 2
! 1
! 2

样例输出 #1

1 1

提示

对于 $100\%$ 的数据， $n\leq 200000$ , $m\leq 500000$

单链表思路解释

考虑维护一个单链表来表示好友关系和收到消息个数，其中每个单链表的头结点指出这条链表表示的是谁的关系。例如对于以下输入有如下链表变化：

+ 1 2
+ 2 3
+ 1 3

第一行执行完毕：

1->2
2->1

第二行执行完毕：

1->2
2->1->3
3->2

第三行执行完毕：

1->2->3
2->1->3
3->2->1

创建一个头结点数组，且头结点中存有用户序号和收到消息数量，在添加好友和删除好友时对相应链表进行增加和删除，则有如下代码：
需要注意的是，本思路及代码运行时间会随着单一用户好友数量显著增加，若许多用户好友数有1w+，则TLE难以避免


#include <iostream>
#include <vector>
using namespace std;
class Node {
public:
    int user;
    int message;
    Node* next;
    Node() {
        message = 0;
        next = NULL;
    }
};
Node user[200001];
void countmessage(int);
void addfriend(int, int);
void removefriend(int, int);
void printall(int n) {
    for (int i = 1; i <= n; i++) {
        cout << user[i].message << " ";
    }
    cout << endl;
}
void addfriend(int a, int b) {//假定添加好友没有重复
    Node* newf = new Node;
    newf->user = b;
    Node* p = user + a;
    while (p->next) {
        p = p->next;
    }
    p->next = newf;
    newf = new Node;
    newf->user = a;
    p = user + b;
    while (p->next) {
        p = p->next;
    }
    p->next = newf;
}
void removefriend(int a, int b) {//假定没有重复的好友关系，假定没有空好友的情况
    Node* p = user + a; Node* prep = p; p = p->next;
    while (p) {
        if (p->user == b) {
            prep->next = p->next;
            delete p;
            break;
        }
        prep = p;
        p = p->next;
    }
    p = user + b; prep = p; p = p->next;
    while (p) {
        if (p->user == a) {
            prep->next = p->next;
            delete p;
            break;
        }
        prep = p;
        p = p->next;
    }
}
void countmessage(int u) {
    Node* p = user + u; p = p->next; 
    while (p) {
        user[p->user].message++;
        p = p->next;
    }
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    int n, m;
    cin >> n; cin >> m;
    for (int i = 0; i < n; i++) {
        user[i].message = 0;
    }
    string s;
    while (m--) {
        cin >> s; int a = 0, b = 0;
        switch (s[0]) {
        case '!': cin >> a; countmessage(a); break;
        case '+': cin >> a; cin >> b; addfriend(a, b); break;
        case '-': cin >> a; cin >> b; removefriend(a, b); break;
        }
        s.clear();
    }
    printall(n);

}