18114 Ly and PostCards 2016scau校賽題目

需要一點思維轉變 就是一個數會被算入期望值必定是當且僅當他或他關聯的數字被選入 所以我們對於一個數只需要遞歸判斷他的關聯數字有哪些 計爲cnt 得到答案就應該是c（n，k）-c（n-cnt，k）

#include <cstdio>
#include <iostream>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <queue>
#include <stack>
#include <bitset>
#include <vector>
#include <string>
#include <deque>
#include <set>
#include <map>
#pragma comment(linker, "/STACK:1024000000,1024000000")
using namespace std;

typedef long long ll;
const int maxn=55;
vector<int>a[maxn];
ll c[maxn+5][maxn+5];
int vis[maxn];
ll cnt[maxn];

void dfs(int cur)
{
    //if(vis[cur]) return;
    vis[cur]=1;
    int n=a[cur].size();
    //cout<<n<<endl;
    for(int i=0;i<n;i++)
    {
        if(!vis[a[cur][i]])
        {
            //vis[a[cur][i]]=1;
            dfs(a[cur][i]);
        }
    }
}
int main()
{
    c[0][0]=1;
    for(int i=0;i<=maxn;i++) c[i][0]=1;
    for(int i=1;i<=maxn;i++)
    {
        for(int j=1;j<=maxn;j++)
        {
            c[i][j]=c[i-1][j-1]+c[i-1][j];
        }
    }
    for(int i=0;i<=maxn;i++) c[0][i]=0;
    /*for(int i=0;i<=20;i++)
    {
        for(int j=0;j<=i;j++)
        {
            printf("%4d ",c[i][j]);
            cout<<i<<' '<<j<<endl;
        }
        cout<<endl;
    }*/
    //cout<<c[2][1]<<endl;
    int t;
    cin>>t;
    while(t--)
    {
        memset(vis,0,sizeof(vis));
        //memset(c,0,sizeof(c));
        memset(cnt,0,sizeof(cnt));
        int n,m,k;
        scanf("%d%d%d",&n,&m,&k);
        for(int i=1;i<=n;i++) a[i].clear();
        for(int i=1;i<=m;i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            a[v].push_back(u);
        }
        ll ans=0;
        for(int i=1;i<=n;i++)
        {
            ll tot=0;
            memset(vis,0,sizeof(vis));
            dfs(i);
            for(int j=1;j<=n;j++) {if(vis[j]) cnt[i]++;}
           // ans+=(c[n][k]-c[n-tot][k]);
           //cout<<cnt[i]<<" jklj "<<endl;
        }
        //cout<<c[2][1]<<"dfdf"<<endl;
        for(int i=1;i<=n;i++)
        {
            //cout<<n<<' '<<k<<endl;
            //cout<<i<<' '<<c[n][k]<<' '<<c[n-cnt[i]][k]<<endl;
            ans+=c[n][k]-c[n-cnt[i]][k];
        }
        printf("%lld\n",ans);
    }
}