A:小美的排列询问
小模拟:用一个map来记录数字的位置,在后面判断两个数的abs(mp[a]-mp[b])是不是等于1就可以
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+10;
char c[1001][1001];
int val[N];//每个点的权值
vector<int>g[N];
bool vis[N];
int check(int x,int y)
{
ll p=x*y;
int u=(int)sqrt(p);
if(u*u==p)return 1;
return 0;
}
int dfs(int u,int fa)
{
int cnt=0;
for(auto v:g[u])
{
if(v==fa)continue;
int w=dfs(v,u);
if(vis[v]==0&&vis[u]==0&&check(val[v],val[u]))
{
//搜vu两个点
vis[u]=vis[v]=1;
cnt+=2;
}
cnt+=w;
}
return cnt;
}
void cf()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>val[i];
}
for(int i=1;i<=n-1;i++)
{
int u,v;
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1;i<=n;i++)
{
if(g[i].size()==1){
int ans=dfs(i,-1);
cout<<ans<<endl;
break;
}
}
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t;
t=1;
while(t--){
cf();
}
}
B题:小美走公路
也是一道模拟题不过终点可能在起点前面这是一个坑点
所有x-y的距离不可以直接用for循环去遍历
while(x!=y){
ans+=a[x];
x=(x+1)%n;
}
这里开了一个while,每次取模的话可以避免x套圈走
然后每种路径都有两种答案一种是x-y的,一种是反向走的
如图所示:所以答案一种是ans一种是sum-ans
我们两个取min就可以了
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
ll a[N];
void cf()
{
ll n;
cin>>n;
ll sum=0;
for(int i=0;i<n;i++){
cin>>a[i];
sum+=a[i];
}
ll x,y;
cin>>x>>y;
x--;
y--;
ll ans=0;
while(x!=y){
ans+=a[x];
x=(x+1)%n;
}
cout<<min(ans,sum-ans);
}
int main()
{
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t;
t=1;
while(t--)
{
cf();
}
}
C题:小美的排序构造:
我们可以很清楚的知道比如5:
我们可以构造出来:1 5 2 4 3
所以我们可以把每两个数之和构造n+1和n+2这样极差就是1
因此每次取出来一个数我们可以输出i和(n+1)-i
i的范围是在1-(n+1)/2这里的n+1是上取整操作
还有一个点是要注意奇偶
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int main(){
int n;
cin>>n;
ll q=0;
q=(n+1)/2;
ll ans=n+1;
if(n&1)
{
for(int i=1;i<=q;i++){
if(i<=q-1){
cout<<i<<" "<<ans-i<<" ";
}
else
{
cout<<i;
}
}
}
else
{
for(int i=1;i<=q;i++)
{
cout<<i<<" "<<ans-i<<" ";
}
}
}
D题:小美的树上染色
有一种贪心的做法是直接找一个叶子然后从叶子开始去模拟遍历
以叶子为父节点一层一层去遍历
这里的check函数是判断乘积是不是完全平方数
vis数组是判断这个点有没有染色
dfs函数的意思是以u为节点fa为u的父节点向下搜的答案
所以w就是以u为父节点的答案
然后我们只需要再判断u和v能不能染色然后把两种答案相加就是总答案
int dfs(int u,int fa)
{
int cnt=0;
for(auto v:g[u])
{
if(v==fa)continue;
int w=dfs(v,u);
if(vis[v]==0&&vis[u]==0&&check(val[v],val[u]))
{
//搜vu两个点
vis[u]=vis[v]=1;
cnt+=2;
}
cnt+=w;
}
return cnt;
}
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=3e5+10;
char c[1001][1001];
int val[N];//每个点的权值
vector<int>g[N];
bool vis[N];
int check(int x,int y)
{
ll p=x*y;
int u=(int)sqrt(p);
if(u*u==p)return 1;
return 0;
}
int dfs(int u,int fa)
{
int cnt=0;
for(auto v:g[u])
{
if(v==fa)continue;
int w=dfs(v,u);
if(vis[v]==0&&vis[u]==0&&check(val[v],val[u]))
{
//搜vu两个点
vis[u]=vis[v]=1;
cnt+=2;
}
cnt+=w;
}
return cnt;
}
void cf()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>val[i];
}
for(int i=1;i<=n-1;i++)
{
int u,v;
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
for(int i=1;i<=n;i++)
{
if(g[i].size()==1){
int ans=dfs(i,-1);
cout<<ans<<endl;
break;
}
}
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t;
t=1;
while(t--){
cf();
}
}