PAT1076. Forwards on Weibo (30)

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了PAT1076. Forwards on Weibo (30)相关的知识,希望对你有一定的参考价值。

使用DFS出现超时,改成bfs

DFS

#include <iostream>
#include <vector>
#include <set>
using namespace std;
int n;
int l;
int i,j;
int m,k;
int follower[1001][1001];
vector<int> user_list[1001];
set<int> user_s;
void queryU(int a,int depth){
  if(depth==l){
    return ;
  }
  for(int i=0;i<user_list[a].size();i++){
    queryU(user_list[a][i],depth+1);
    user_s.insert(user_list[a][i]);
  }
  
  return ;
}
int main()
{
  cin>>n>>l;
  int a;
  for(i=1;i<=n;i++){
    cin>>m;
    for(j=0;j<m;j++){
      cin>>a;
      user_list[a].push_back(i);
    }
  }
  cin>>k;
  int sum;
  for(i=0;i<k;i++){
    cin>>a;
    user_s.clear();
    user_s.insert(a);
    queryU(a,0);
    cout<<user_s.size()-1<<endl;
  }
    return 0;
}

  bfs

最后一个测试用例一直MLE,后来用指针优化,就报超时了。

后来参考 https://www.zhihu.com/question/28264519,把队列删除语句省掉,set省掉,终于AC了!

#include <iostream>
#include <vector>
#include <set>
using namespace std;
int n;
int l;
int i,j;
int m,k;
vector<int> user_list[1010];
vector<int> que;
int v[1005];
int main()
{
	cin>>n>>l;
	int a;
	for(i=1;i<=n;i++){
		cin>>m;
		for(j=0;j<m;j++){
			cin>>a;
			user_list[a].push_back(i);
		}
	}
	cin>>k;
	for(i=1;i<=k;i++){
		cin>>a;
		v[a]=i;
		int depth=0;
		que.clear();
		que.push_back(a);
		int levelSize=0;
		int count=1;
		for(int z=0;z<que.size();z++){
			int num=que[z];
			count--;
			vector<int> *tmp=&user_list[num];
			levelSize+=tmp->size();
			for(int y=0;y<tmp->size();y++){
				int temp_num=(*tmp)[y];
				if(v[temp_num]!=i){
					que.push_back(temp_num);
					v[temp_num]=i;
				}
			}
			if(count==0){
				count=levelSize;
				levelSize=0;
				depth++;
			}
			if(depth==l){
				break;
			}
		}
		cout<<que.size()-1<<endl;
	}
	return 0;
}



	

  

 

以上是关于PAT1076. Forwards on Weibo (30)的主要内容,如果未能解决你的问题,请参考以下文章

PAT A1076 Forwards on Weibo (30 分)——图的bfs

1076. Forwards on Weibo (30)树+搜索——PAT (Advanced Level) Practise

PAT1076. Forwards on Weibo(标准bfs模板)

PAT Advanced 1076 Forwards on Weibo (30) [图的遍历,BFS,DFS]

PAT (Advanced Level) 1076. Forwards on Weibo (30)

1076. Forwards on Weibo (30)