笔面试算法--常用模板介绍（爱丽丝的人偶(二) (选做）、点一成零(选做)、maki和tree (选做)）

本文链接：https://blog.csdn.net/weixin_44170305/article/details/123607617

本文介绍了三个C++程序，分别涉及组合计数、二维网格图的连通块计算和有向图联通块的并查集实现。第一个程序计算特定数量元素的组合数；第二个程序处理二维网格中1表示的连通区域，并在修改网格后更新连通块信息；第三个程序关注有向图中节点的联通性，并计算联通块的加权乘积。这些程序展示了组合数学和图论在算法中的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include<cstdio>
#include<cmath>
#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<iomanip>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<limits.h>
#include<set>
#include<cstring>
#include<string>
#include<numeric>
#include <typeinfo>
#define inf 0x3f3f3f3f
#define wc 1e-4
#define PI 3.1415926536
#define MAX(a,b) ((a)>(b)?(a):(b))
using namespace std;
typedef long long ll;
const int mod=1e9+7;
int n,k;
set<ll>a;
//int f[100010];
//void init(){
//    for(int i=0;i<=n;i++)
//        f[i]=i;
//}
//int fd(int x){
//    return f[x]==x?x:f[x]=fd(f[x]);
//}
//void hebin(int x,int y){
//    int xx=fd(x);
//    int yy=fd(y);
//    if(xx!=yy)
//        fa[xx]=yy;
//}
ll kpow(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1)
            ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
ll inv(ll x){
    return kpow(x,mod-2);
}
ll C(ll n,ll m){
    ll ans=1;
    for(ll i=1;i<=m;i++){
        ans*=(n-i+1);
        ans%=mod;
        ans*=inv(i);
        ans%=mod;
    }
    return ans;
}
int main(){
    cin>>n>>k;
    for(int i=1;i<=n;i++){
        ll x;
        cin>>x;
        a.insert(x);
    }
    int cd=a.size();
    if(cd<k)
        cout<<0<<endl;
    else
        cout<<C(cd,k)<<endl;
    return 0;
}

#include<iostream>
#include<cstdio>
#include<cmath>
#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<iomanip>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<limits.h>
#include<set>
#include<cstring>
#include<string>
#include<numeric>
#include <typeinfo>
#define inf 0x3f3f3f3f
#define wc 1e-4
#define PI 3.1415926536
#define MAX(a,b) ((a)>(b)?(a):(b))
using namespace std;
typedef long long ll;
const int mod=1e9+7;
int f[250010],s[250010];
int n,k;
char a[510][510];
ll p[250010];
int dx[]= {0,0,-1,1};
int dy[]= {-1,1,0,0};
int fd(int x)
{
    return f[x]==x?x:f[x]=fd(f[x]);
}
void hebin(int x,int y)
{
    int xx=fd(x);
    int yy=fd(y);
    if(xx==yy)
        return ;
    if(s[xx]>s[yy]) //个数少的合并到个数多的连通块
    {
        s[xx]+=s[yy];
        f[yy]=xx;
    }
    else
    {
        s[yy]+=s[xx];
        f[xx]=yy;
    }
}
int calc(int x,int y) //二维化为一维
{
    return (x-1)*n+y;
}
ll kpow(ll a,ll b)
{
    ll ans=1;
    while(b)
    {
        if(b&1)
            ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
ll inv(ll x)
{
    return kpow(x,mod-2);
}
int main()
{
    cin>>n;
    p[0]=1;
    for(int i=1; i<=n*n; i++)
    {
        f[i]=i;
        s[i]=1;
        p[i]=p[i-1]*i%mod;
    }
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            cin>>a[i][j];
    for(int i = 1; i <= n; i ++)
    {
        for(int j = 1; j <= n; j ++)
        {
            if(a[i][j] == '0')
                continue;
            if(j + 1 <= n && a[i][j + 1] == '1')
                hebin(calc(i, j), calc(i, j + 1));
            if(i + 1 <= n && a[i + 1][j] == '1')
                hebin(calc(i, j), calc(i + 1, j));
        }
    }
    //ans是连通块大小的乘积，ct是连通块个数
    ll ans = 1, ct = 0;
    for(int i = 1; i <= n; i ++)
    {
        for(int j = 1; j <= n; j ++)
        {
            int t = calc(i, j);
            if(a[i][j] == '1' && fd(t) == t)
            {
                ans = ans * s[t] % mod;
                ct ++;
            }
        }
    }
    cin>>k;
    while(k--)
    {
        int x,y;
        cin>>x>>y;
        x++;
        y++;
        if(a[x][y]=='1')
        {
            cout<<ans*p[ct]%mod<<endl;
            continue;
        }
        a[x][y]='1';
        ct++;
        for(int i=0; i<4; i++)
        {
            int tx=x+dx[i];
            int ty=y+dy[i];
            if(tx>=1&&tx<=n&&ty>=1&&ty<=n&&a[tx][ty]=='1')
            {
                int t1=fd(calc(x, y));
                int t2=fd(calc(tx, ty));
                if(t1!=t2)//合并两个连通块
                {
                    ans=ans*inv(s[t1])%mod*inv(s[t2])%mod*(s[t1]+s[t2])%mod;
                    ct--;
                    hebin(t1, t2);
                }
            }
        }
        cout<<ans*p[ct]%mod<<endl;
    }
    return 0;
}

#include<iostream>
#include<cstdio>
#include<cmath>
#include<stdio.h>
#include<stdlib.h>
#include<vector>
#include<iomanip>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
#include<limits.h>
#include<set>
#include<cstring>
#include<string>
#include<numeric>
#include <typeinfo>
#define inf 0x3f3f3f3f
#define wc 1e-4
#define PI 3.1415926536
#define MAX(a,b) ((a)>(b)?(a):(b))
using namespace std;
typedef long long ll;
const int mod=1e9+7;
int f[100010],km[100010],t[100010];
int n;
string s;
vector<int>mp[100010];
int fd(int x)
{
    return f[x]==x?x:f[x]=fd(f[x]);
}
void hebin(int x,int y)
{
    int xx=fd(x);
    int yy=fd(y);
    if(xx!=yy)
    {
        if(km[xx]<km[yy])
        {
            f[xx]=yy;
            km[yy]+=km[xx]+1;

        }
        else
        {
            f[yy]=xx;
            km[xx]+=km[yy]+1;
        }
    }
}
ll gcode(vector<int> tmp)
{
    ll ans=0;
    int n=tmp.size();
    if (n==0)
        return 0;
    ll dp[100010]= {0}, sum[100010]= {0}, s=0;
    for (int i=0; i<n; i++)
        ans+=tmp[i];
    for (int i=0; i<n; i++)
    {
        s+=tmp[i];
        sum[i]=s;
    }
    for (int i=1; i<n; i++)
        dp[i]=dp[i-1]+tmp[i]*sum[i-1];
    return ans+dp[n-1];
}
int main()
{
    cin>>n;
    cin>>s;
    for(int i=1; i<=n; i++)
        f[i]=i;
    for(int i=1; i<n; i++)
    {
        int x,y;
        cin>>x>>y;
        mp[x].push_back(y);
        mp[y].push_back(x);
        if(s[x-1]=='W'&&s[y-1]=='W')
            hebin(x,y);
    }
    for (int i=1; i<=n; i++)
        t[i]=km[fd(i)]+1;
    ll sum=0;
    for (int i=1; i<=n; i++)
    {
        if (s[i-1]=='B')
        {
            vector<int> tmp;
            for (int j=0; j<mp[i].size(); j++)
            {
                if (s[mp[i][j]-1]=='W')
                    tmp.push_back(t[mp[i][j]]);
            }
            sum+=gcode(tmp);
        }
    }
    cout<<sum<<endl;
    return 0;
}