题目
思路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;
}