模拟题
STL题
L3-002 特殊堆栈（两个vector）
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1e5+10;
int n;
vector<int>v,v1;//v有序
int main()
{
cin>>n;
while(n--)
{
string a;cin>>a;
if(a=="Pop")
{
if(v1.empty()) puts("Invalid");
else
{
int t=v1.back();
auto it=lower_bound(v.begin(),v.end(),t);
v.erase(it);
cout<<t<<endl;
v1.pop_back();
}
}
else if(a=="PeekMedian")
{
if(v1.empty()) puts("Invalid");
else cout<<v[(v.size()+1)/2-1]<<endl;
}
else
{
int t;cin>>t;
v1.pb(t);
auto it=lower_bound(v.begin(),v.end(),t);
v.insert(it,t);
}
}
return 0;
}
dp题
搜索题
L3-001 凑零钱（dfs）
dfs：从小到大排列，找到的第一个答案就是序列最小的答案。如果从大到小排列会TLE两个点。
也可以dp（01背包）。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1e4+10;
int n,m,a[N],sum,fl;
vector<int>ans,anss;
void dfs(int u,int sum)
{
if(fl) return;
if(sum>m) return;
if(sum==m)
{
fl=1;
anss=ans;
return;
}
for(int i=u;i<=n;i++)
{
ans.pb(a[i]);
dfs(i+1,sum+a[i]);
ans.pop_back();
}
}
int main()
{
cin>>n>>m;
fir(i,1,n)
{
cin>>a[i];sum+=a[i];
}
if(sum<m)
{
cout<<"No Solution";return 0;
}
sort(a+1,a+1+n);
dfs(1,0);
if(fl)
{
int f=0;
for(auto x:anss)
{
if(f) cout<<" ";
cout<<x;f++;
}
}
else cout<<"No Solution";
return 0;
}
L3-004 肿瘤诊断（bfs）
三维bfs。
dfs会段错误，估计是递归次数太多了。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1500+10;
int g[1300][130][80],v[1300][130][80];
int n,m,l,t;
int ans,temp;
int dx[6]={0,0,0,0,1,-1};
int dy[6]={0,0,1,-1,0,0};
int dz[6]={1,-1,0,0,0,0};
struct node
{
int x,y,z;
};
void bfs(int x,int y,int z)
{
queue<node>q;
q.push({x,y,z});
while(q.size())
{
node t=q.front();q.pop();
for(int i=0;i<6;i++)
{
int xx=t.x+dx[i];
int yy=t.y+dy[i];
int zz=t.z+dz[i];
if(xx>=1&&xx<=m&&yy>=1&&yy<=n&&zz>=1&&zz<=l)
{
if(!v[xx][yy][zz]&&g[xx][yy][zz])
{
v[xx][yy][zz]=1;
temp++;
q.push({xx,yy,zz});
}
}
}
}
}
int main()
{
cin>>m>>n>>l>>t;
fir(i,1,l)
fir(j,1,m)
fir(k,1,n)
cin>>g[j][k][i];
fir(i,1,l)
fir(j,1,m)
fir(k,1,n)
{
if(!v[j][k][i]&&g[j][k][i])
{
temp=1;
v[j][k][i]=1;
bfs(j,k,i);
if(temp>=t) ans+=temp;
}
}
cout<<ans;
return 0;
}
L3-008 喊山（bfs）
没懂“题目假设每个山头最多有两个能听到它的临近山头”的意思，直接模板bfs一下。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1e4+10;
int n,m,k;
vector<int>g[N];
struct node
{
int a,ans;
};
int id,maxn;
int v[N];
int bfs(int x)
{
memset(v,0,sizeof(v));
queue<node>q;
q.push({x,0});
v[x]=1;
id=-1,maxn=-1;
while(q.size())
{
node t=q.front();q.pop();
for(auto u:g[t.a])
{
if(v[u]) continue;
v[u]=1;
node temp={u,t.ans+1};
if(t.ans+1>maxn)
{
maxn=t.ans+1;
id=u;
}
else if(t.ans+1==maxn&&u<id)
{
id=u;
}
q.push(temp);
}
}
return id;
}
int main()
{
cin>>n>>m>>k;
fir(i,1,m)
{
int a,b;cin>>a>>b;
g[a].pb(b);g[b].pb(a);
}
while(k--)
{
int kk;cin>>kk;
int ans=bfs(kk);
if(ans==-1) puts("0");
else cout<<ans<<endl;
}
return 0;
}
图论题
L3-011 直捣黄龙（dij+dfs+map）
要素过多的大杂烩属于是。
听说直接dfs也可以满分，数据有点水。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define ll long long
#define pb push_back
const int N=200+10;
int n,m;
string start,endd;
map<string,int>mp;
map<int,string>mpp;
int idx=1;
void add(string a)
{
mp[a]=idx;
mpp[idx]=a;
idx++;
}
int p[N];//people
int g[N][N];
int ss,ee;//start end
int dist[N],st[N];
int dij()
{
memset(dist,0x3f,sizeof(dist));
dist[ss]=0;
for(int i=1;i<=n;i++)
{
int t=-1;
for(int j=1;j<=n;j++)
{
if(!st[j]&&(t==-1||dist[j]<dist[t]))
{
t=j;
}
}
st[t]=1;
for(int j=1;j<=n;j++)
{
dist[j]=min(dist[j],dist[t]+g[t][j]);
}
}
return dist[ee];
}
int minn,all,killl;
vector<int>ans,temp;
int v[N];
void dfs(int u,int kill)
{
if(u==ee)
{
all++;
if(temp.size()>ans.size())
{
ans=temp;killl=kill;
}
else if(temp.size()==ans.size()&&kill>killl)
{
killl=max(killl,kill);ans=temp;
}
return;
}
for(int i=1;i<=n;i++)
{
if(!v[i]&&dist[i]==dist[u]+g[u][i])
{
v[i]=1;
temp.pb(i);
dfs(i,kill+p[i]);
temp.pop_back();
v[i]=0;
}
}
}
int main()
{
cin>>n>>m>>start>>endd;
add(start);add(endd);
fir(i,1,n-1)
{
string a;int b;cin>>a>>b;
if(mp.find(a)==mp.end())
{
add(a);
}
int t=mp[a];p[t]=b;
}
memset(g,0x3f,sizeof(g));
fir(i,1,m)
{
string a,b;int c;cin>>a>>b>>c;
int t1=mp[a],t2=mp[b];
g[t1][t2]=g[t2][t1]=c;
}
ss=mp[start],ee=mp[endd];
//最短路的长度
minn=dij();
all=0;ans.clear();
killl=0;
v[ss]=1;
temp.pb(ss);
dfs(ss,0);
int f=0;
for(auto x:ans)
{
if(f) cout<<"->";
cout<<mpp[x];f++;
}
cout<<endl;
cout<<all<<" "<<minn<<" "<<killl;
return 0;
}
L3-005 垃圾箱分布（dij）
对每个垃圾桶dij。
注意：最后输出要lf，如果只是f就会最后一点过不了。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1e3+30;
int n,m,k,ds;
int toint(string a)
{
int anss=0;
for(int i=0;a[i];i++)
{
if(a[i]>='0'&&a[i]<='9') anss=anss*10+a[i]-'0';
}
return anss;
}
int g[N][N];
int dist[N],st[N];
void dij(int s)//计算所有点到s的最短路
{
memset(dist,0x3f,sizeof(dist));
memset(st,0,sizeof(st));
dist[s]=0;//s点为起点
for(int i=1;i<=n+m;i++)
{
int t=-1;
for(int j=1;j<=n+m;j++)
if(!st[j]&&(t==-1||dist[j]<dist[t]))
t=j;
st[t]=1;
for(int j=1;j<=n+m;j++)
dist[j]=min(dist[j],dist[t]+g[t][j]);
}
}
int ansid;double ans1,ans2;
int main()
{
cin>>n>>m>>k>>ds;
memset(g,0x3f,sizeof(g));
fir(i,1,k)
{
string a,b;int c;cin>>a>>b>>c;
int aa=toint(a),bb=toint(b);
if(a[0]=='G') aa+=n;
if(b[0]=='G') bb+=n;
g[aa][bb]=g[bb][aa]=c;
}
ansid=11;ans1=-1,ans2=0x3f3f3f3f;
for(int i=1;i<=m;i++)
{
int t=i+n;
dij(t);
int minn=0x3f3f3f3f;double pj=0;int f=0;
for(int j=1;j<=n;j++)
{
if(dist[j]>ds)
{
f=1;break;
}
minn=min(dist[j],minn);
pj+=dist[j];
}
pj/=n;
if(!f)
{
int flag=0;
if(minn>ans1) flag++;
else if(ans1==minn&&pj<ans2) flag++;
else if(ans1==minn&&ans2==pj&&i<ansid) flag++;
if(flag)
{
ansid=i;ans1=minn;ans2=pj;
}
}
}
if(ansid==11) puts("No Solution");
else
{
printf("G%d\n",ansid);
printf("%.1lf %.1lf",(double)ans1,ans2);//不是lf就会最后一个点过不了 扣4分
}
return 0;
}
二叉树题
L3-010 是否完全二叉搜索树
注意：如果不是完全二叉树，那么有可能结点的下标非常大，可能会爆int或LL，所以要用map。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=20+10;
map<int,int>mp;
int n;
void in(int x)
{
int xx=1;
while(mp.find(xx)!=mp.end())
{
if(x>mp[xx]) xx*=2;
else xx=xx*2+1;
}
mp[xx]=x;
}
int main()
{
cin>>n;
cin>>mp[1];
fir(i,2,n)
{
int t;cin>>t;
in(t);
}
vector<int>temp;
int f=0,ff=0;
for(auto x:mp)
{
if(f) cout<<" ";
f++;cout<<x.second;
temp.pb(x.first);
}
for(int i=0;i<temp.size()-1;i++)
{
if(temp[i]+1!=temp[i+1]) ff=1;
}
cout<<endl;
if(ff) cout<<"NO";
else cout<<"YES";
return 0;
}
L3-016 二叉搜索树的结构（二叉树建树+map+模拟）
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1e5+10;
int n,m;
map<int,int>mp,ans;
void ins(int x)
{
int xx=1;
while(mp.find(xx)!=mp.end())
{
if(x>mp[xx]) xx=xx*2+1;
else xx*=2;
}
mp[xx]=x;ans[x]=xx;
}
int main()
{
cin>>n;
cin>>mp[1];ans[mp[1]]=1;
fir(i,2,n)
{
int t;cin>>t;
ins(t);
}
cin>>m;
while(m--)
{
int a;cin>>a;string s1;cin>>s1;
if(s1=="is")
{
string s2;cin>>s2>>s2;
//the root
if(s2=="root")
{
if(ans.find(a)!=ans.end()&&ans[a]==1) puts("Yes");
else puts("No");
}
//the parent of B
else if(s2=="parent")
{
cin>>s2;int b;cin>>b;
if(ans.find(a)!=ans.end()&&ans.find(b)!=ans.end())
{
int aa=ans[a],bb=ans[b];
if(bb/2==aa) puts("Yes");
else puts("No");
}
else puts("No");
}
//the left child of B
else if(s2=="left")
{
cin>>s2>>s2;int b;cin>>b;
if(ans.find(a)!=ans.end()&&ans.find(b)!=ans.end())
{
int aa=ans[a],bb=ans[b];
if(aa==bb*2) puts("Yes");
else puts("No");
}
else puts("No");
}
//the right child of B
else if(s2=="right")
{
cin>>s2>>s2;int b;cin>>b;
if(ans.find(a)!=ans.end()&&ans.find(b)!=ans.end())
{
int aa=ans[a],bb=ans[b];
if(aa==bb*2+1) puts("Yes");
else puts("No");
}
else puts("No");
}
}
else
{
int b;string s2;
cin>>b>>s2>>s2;
//and B are siblings
if(s2=="siblings")
{
int aa=-1,bb=-1;
if(ans.find(a)!=ans.end()) aa=ans[a];
if(ans.find(b)!=ans.end()) bb=ans[b];
if(aa==bb+1||bb==aa+1) puts("Yes");
else puts("No");
}
//and B are on the same level
else
{
int aa=0,bb=0,f=0;
if(ans.find(a)!=ans.end()) a=ans[a];
else f++;
if(ans.find(b)!=ans.end()) b=ans[b];
else f++;
while(a)
{
a/=2;aa++;
}
while(b)
{
b/=2;bb++;
}
if(aa==bb&&!f) puts("Yes");
else puts("No");
string temp;
fir(i,1,3) cin>>temp;
}
}
}
return 0;
}
并查集题
L3-003 社交集群
注意，这里问的是每个并查集内的人数，因此要用每个人的第一个爱好来代表这个人。
#include<bits/stdc++.h>
using namespace std;
#define fir(i,a,n) for(int i=a;i<=n;i++)
#define mem(a,x) memset(a,x,sizeof(a));
#define pb push_back
typedef long long ll;
const int N=1e3+10;
int n,p[N];
void ini()
{
for(int i=1;i<=1000;i++) p[i]=i;
}
int find(int x)
{
while(x!=p[x]) x=p[x];
return p[x];
}
void merge(int a,int b)
{
int aa=find(a),bb=find(b);
if(aa!=bb) p[bb]=aa;
}
int v[N],a[N];
bool cmp(int a,int b)
{
return a>b;
}
int main()
{
ini();
cin>>n;
fir(i,1,n)
{
int t;cin>>t;char op;cin>>op;
cin>>a[i];t--;
while(t--)
{
int temp;cin>>temp;
merge(a[i],temp);
}
}
map<int,int>mp;
fir(i,1,n)
{
int fa=find(a[i]);
mp[fa]++;
}
cout<<mp.size()<<endl;
vector<int>ans;
for(auto x:mp) ans.pb(x.second);
sort(ans.begin(),ans.end(),cmp);
int f=0;
for(auto x:ans)
{
if(f) cout<<" ";
f++;cout<<x;
}
return 0;
}
1070
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
