#C. 笨友圈

题目

---

思路1

直接模拟题意,相当于邻接矩阵,用bool类型的二维数组vis[i][j]来存储i和j是否为好友,交叉点为1代表是好友,为0代表不是;a[i]存储i这个人看到的信息数量。

然后输入后如果符号代表要将u,v加个好友,就将vis[u][v] = 1和vis[v][u] = 1,如果是拉黑就将vis[u][v] = 0和vis[v][u] = 0,否则就是发信息，那么就将所以a[j]++(满足u,j是好友,即vis[u][j] == 1)。

代码1

#include<bits/stdc++.h>
using namespace std;
#define int long long 
int n,m,a[1000000];
bool vis[20001][20001];
signed main()
{
  cin>>n>>m;
  for(int i = 0;i < m;i++)
  {
      char t;
      int u,v;
      cin>>t;
    if(t == '+')
    {
      cin>>u>>v;
      vis[u][v] = 1;
      vis[v][u] = 1;
    }
    else if(t == '-')
    {
      cin>>u>>v;
      vis[u][v] = 0;
      vis[v][u] = 0;
    }
    else if(t == '!')
      {
        cin>>u;
        for(int j = 1;j <= n;j++)
          if(vis[u][j] == 1)
            a[j]++;
    }
  }
  for(int i = 1;i <= n;i++) cout<<a[i]<<" ";
  return 0;
}

分析1

为什么会RE呢?因为数据范围是

,所以vis[][]因开为vis[200000][200000]。可是这样的会又会导致爆空间。(4^10哪!)所以该怎么办呢?

---

思路2

为了避免爆空间,可以用一个set<int> vec[1000001],用类似于邻接表的方式来存加好友的关系。

并且可以用p[i]来存i能看到的消息条数,a[i]存i发的消息条数。

其他细节详见注释

代码2

#include<bits/stdc++.h>
using namespace std;
#define int long long
int n,m,a[1000001],p[1000001];
set<int> vec[1000001];
signed main()
{
  cin>>n>>m;
  for(int i = 0; i < m; i++)
  {
    char t;
    int u,v;
    cin>>t;
    if(t == '+')
    {
      cin>>u>>v;
      vec[u].insert(v);//互加好友
      vec[v].insert(u);
      p[u] -= a[v];
      p[v] -= a[u];
    }
    else if(t == '-')
    {
      cin>>u>>v;
      auto ta = vec[u].find(v),tb = vec[v].find(u);
      //ta代表在u的邻接点中v的地址,tb同理
      vec[u].erase(ta);//互删好友
      vec[v].erase(tb);
      p[u] += a[v];
      p[v] += a[u];
    }
    else if(t == '!')
    {
      cin>>u;
      a[u]++;//u发的消息条数+1
    }
  }
  for(int i = 1; i <= n; i++)
    for(auto j = vec[i].begin(); j != vec[i].end(); j++)//遍历i的临接点的地址
      p[*j] += a[i];//将j能看到的消息条数+=i发的条数(因为他两互为好友)
  for(int i = 1; i <= n; i++) cout<<p[i]<<" ";//输出i能看到的消息条数
  return 0;
}