A.并查集
比较基本的并查集操作,按照题目要求来就可以了
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N = 4e6 + 7;
const ll M = 8e6 + 7;
const ll mod = 998244353;
ll fa[N];
ll find(ll x) {
return x == fa[x] ? fa[x] : fa[x] = find(fa[x]);
}
void in(ll x, ll y) {
ll xx = find(x), yy = find(y);
if (xx != yy)
fa[xx] = yy;
}
bool cmp(ll x, ll y) {
if (find(x) == find(y))
return true;
else
return false;
}
void init() {
for (ll i = 0; i <= N; i++)
fa[i] = i;
}
int quick(ll x, ll y) {
ll ans = 1;
while (y) {
if (y & 1) {
ans = (ans * x) % mod;
}
x = (x * x) % mod;
y = (y >> 1) % mod;
}
return ans % mod;
}
int main() {
ll n, m;
cin >> n >> m;
vector<ll>q;
ll op, x, y;
init();
ll sum = 0;
while (m--) {
//cin>>op>>x>>y;
scanf("%lld%lld%lld", &op, &x, &y);
if (op == 0) {
in(x, y);
} else if (op == 1) {
if (cmp(x, y))
q.push_back(1);
else
q.push_back(0);
}
}
for (ll i = 0; i < q.size(); i++) {
if (q[i] == 1) {
sum = (sum + quick(2, q.size() - i - 1)) % mod;
}
}
cout << sum % mod << endl;
return 0;
}
B.数星星Starts
就是一个树状数组问题
思路:
对于每个星星按y坐标从小到大排序,相同y坐标按x坐标从小到大排序。输入顺序已排好序,那么只要依次统计星星i之前x坐标小于等于i.x的星星有多少,即是星星i的级别。y坐标没用,显然是一维树状数组应用。
设数组a[x]初始为0,表示在x坐标处星星的个数,则求和a[0] ~ a[x]则为该星星的等级。
有一个地方尤其要注意,树状数组是以1号为起始的,而且只能用1号。x可能为0,为0会时陷入死循环,处理时要将所有的x+1。
还有就是x的范围不能事先确定,在update的时候直接加到x取值范围的最大值,即32001,或者在输入的时候找一个x的最大值。
#include<iostream>
#include<cstdio>
using namespace std;
int n,x,y,s,a[32005],b[32005];
int cx(int x)//查询
{
s=0;//初值
for(;x;x-=x&(-x))s+=a[x];//累加区间的值
return s;//返回
}
void update(int x,int y)//更改
{
for(;x<=32001;x+=x&(-x))a[x]+=y;//更改
}
int main()
{
scanf("%d",&n);//用scanf和printf,免得超时
for(int i=1;i<=n;i++)
{
scanf("%d%d",&x,&y);
b[cx(x+1)]++;//累加
update(x+1,1);//更改
}
for(int i=1;i<=n;i++)
printf("%d\n",b[i-1]);
}
C.繁忙的都市
最小生成树裸题,就是建一个最小生成树,然后求该生成树的最大边。
算法1,prim
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 310;
int dist[N];
int g[N][N];
int n,m;
bool st[N];
int prim()
{
memset(dist,0x3f,sizeof dist);
dist[1] = 0;
int res = 0;
for(int i = 0 ;i < n ; i ++)
{
int t = -1;
for(int j = 1 ;j <= n ; j ++)
if(!st[j] && (t == -1 || dist[t] > dist[j]))
t = j;
st[t] = true;
res = max(res,dist[t]);
for(int j = 1 ; j <= n ; j ++)
dist[j] = min(dist[j],g[t][j]);
}
return res;
}
int main()
{
cin>>n>>m;
memset(g,0x3f,sizeof g);
for(int i = 0 ;i < m ; i ++)
{
int a,b,c;
cin>>a>>b>>c;
g[a][b] = g[b][a] = min(g[a][b],c);
}
cout<<n - 1<<" "<<prim()<<endl;
return 0;
}
算法2,Kruskal
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 310,M = 8010;
struct edge{
int a,b,w;
bool operator <(const edge&t)const
{
return w < t.w;
}
}e[M];
int n,m;
int p[N];
int find(int x)
{
if(p[x] != x)p[x] = find(p[x]);
return p[x];
}
int main()
{
cin>>n>>m;
for(int i = 0 ; i < n ; i ++)p[i] = i;
for(int i = 0 ;i < m ; i ++)
{
int a,b,w;
cin>>a>>b>>w;
e[i] = {a,b,w};
}
sort(e,e + m);
int res = 0;
for(int i = 0 ; i < m ; i ++)
{
int a = find(e[i].a),b = find(e[i].b),w = e[i].w;
if(a != b)
{
p[a] = b;
res = max(res,w);//可以直接写res = w(kruskal的性质)
}
}
cout<<n - 1<<" "<<res<<endl;
return 0;
}