2023年码蹄杯二赛 文章压缩 题型:字符串类题目

51 阅读1分钟

码题集OJ-文章压缩 (matiji.net)

思想

我刚开始想用set的count的特性来写,用一个PII类型的set。其中first用来存放字符串,second用来存放字符串的下标。 但是如果是PII类型的set,在适用count的时候也必须查找完全匹配{string,idx},而我只想查找与set里已经存放的string相同的string,而不想查找与其idx也一样的string。

#include<bits/stdc++.h>

using namespace std;
typedef pair<string,int>PII;
set<PII>se;
vector<int>v;
int main()
{
	int n;cin>>n;
    
    for(int i=1;i<=n;i++)
    {
    	string s;cin>>s;
    	
    	if(!se.count(s))
	    {
	    	se.insert({s,i});
		}
		else
		{
			se.insert({s,i});
		}
	}
	
	return 0;
}

后来看了评论区,别人用的map和vector解决了这个问题:

#include<bits/stdc++.h>
using namespace std;
int n;
unordered_map<string,vector<int>>mp;
vector<string>v; 
int main()
{
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		string s;cin>>s;
		
		if(mp[s].empty())v.push_back(s);
		
		mp[s].push_back(i);
		
		
	}
	
	for(int i=0;i<v.size();i++)
	{
		 cout<<v[i]<<"(";
		for(int j=0;j<mp[v[i]].size();j++)   //map.size()返回的map中存的键值对的数量,这里就是v[i]的个数,也就是和v[i]相同的字符串的个数
		{
		 
		   if(!j) cout<<mp[v[i]][j];	
		   else cout<<","<<mp[v[i]][j];

		} 	cout<<")";
	}
	
	return 0;
}

image.png