POJ 3660

len[]数组开小了，以后多开10个

int 最大值 21亿 INF 不能开太大，会溢出

#include <iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int INF =  1 << 10;
int main()
{
    int a,b,N,M,dist[101][101],len1[101],len2[101];
    memset(len1,0,sizeof(len1));
    memset(len2,0,sizeof(len2));
    scanf("%d%d",&N,&M);
    for(int i = 1; i <= N; i++){
        for(int j = 1; j <= N; j++){
            dist[i][j] = INF;
        }
    }
    for(int i = 1; i <= N; i++){
            dist[i][i] = 0;
    }
    for(int i =1; i <= M; i++){
        scanf("%d%d",&a,&b);
        dist[a][b] = 1;
    }
    for(int k = 1; k <= N; k++){
        for(int i = 1; i <= N; i++){
            for(int j = 1; j <= N; j++){
                dist[i][j] = min(dist[i][j],dist[i][k] + dist[k][j]);
            }
        }
    }
    for(int i = 1; i <= N; i++){
        for(int j = 1; j <= N; j++){
            if(dist[i][j] != INF && dist[i][j] > 0){
                len2[i] ++;
            }
            if(dist[j][i] != INF && dist[j][i] > 0){
                len1[i] ++;
            }
        }
    }
    int cnt = 0;
    for(int i = 1; i <= N; i++){
        if(len1[i] + len2[i] == N - 1)
           cnt++;
    }
     printf("%d\n",cnt);
    return 0;
}