智乃的数据库（STL+小技巧）

Posted 2022-02-16 MangataTS

题面连接

https://ac.nowcoder.com/acm/contest/23478/L

题面

思路

• 对于字段名我们直接字符串存储，然后通过map映射到数字，后面方便我们查询
• 然后我们开一个二维数组存储每个数据
• 通过格式化输入来处理SELECT语句
• 最后再通过一个map映射每个字段最后输出聚合的记录数目即可
  详情请看代码

代码

#include<bits/stdc++.h>
using namespace std;
//----------------自定义代码----------------
#define ll long long
#define mod 1000000007
#define endl "\\n"
#define PII pair<int,int>

int dx[4]=0,-1,0,1,dy[4]=-1,0,1,0;

ll ksm(ll a,ll b) 
	ll ans = 1;
	for(;b;b>>=1LL) 
		if(b & 1) ans = ans * a % mod;
		a = a * a % mod;
	
	return ans;


ll lowbit(ll x)return -x & x;

const int N = 1e3+10;
//----------------自定义代码----------------
int n,m,q,a[N][N];
char s[N][55];
map<string,int> mark;
map<vector<int>,int> vis;
vector<int> V,ans;
string temp;
char opt[N*50];

int main()

	scanf("%d %d",&n,&m);
	for(int i = 1;i <= m; i++) 
		scanf(" %s",s[i]);
		mark[s[i]] = i;
	
	for(int i = 1;i <= n; ++i)
		for(int j = 1;j <= m; ++j)	
			scanf("%d",&a[i][j]);
	
	scanf(" SELECT COUNT(*) FROM Table GROUP BY %s",opt);
	for(int i = 0,l = strlen(opt);i < l; ++i)
		if(opt[i] == ',' || opt[i] == ';')
			V.push_back(mark[temp]);
			temp="";
		
		else
			temp += opt[i];
	
		
	for(int i = 1,l = V.size();i <= n; ++i) 
		for(int j = 0;j < l; ++j) 
			ans.push_back(a[i][V[j]]);
		
		vis[ans]++;ans.clear();
	
	printf("%d\\n",vis.size());
	for(auto [kk,pp] : vis)
		printf("%d ",pp);
	
	return 0;